2">&quot;butterfly&quot;</span><span class="p">,</span> <span class="s2">&quot;squirrel&quot;</span><span class="p">])</span>
</pre></div>
</div>
<p>We imported the class custom_segmentation, the class for performing inference and created an instance of the class. We called the model configuration and introduced an extra parameter class_names.</p>
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">class_names</span><span class="o">=</span> <span class="p">[</s