utton><pre><code class="language-python">ClassificationAgent<span class="token punctuation">(</span>classes<span class="token operator">=</span><span class="token punctuation">[</span><span class="token string">'hard hat'</span><span class="token punctuation">,</span> <span class="token string">'no hard hat'</span><span class="token punctuation">]</span><span class="token punctuation">,</span> model<span class="token operator">=</span>CLIP<span class="token punctuation">)</span>
</code></pre></div>
</span><