Update app.py

app.py

@@ -85,7 +85,7 @@ reset_cls_test(predictor.model, classifier, num_classes)
 
 
 
-def inference(img):
+def inference(img,unique_only):
 
     im = cv2.imread(img)
 
@@ -113,7 +113,7 @@ def inference(img):
                 "h": int(height),
             }
         )
-        if (predicted_label not in unique_object_dict):
+        if ((not unique_only) or  (unqiue_only and predicted_label not in unique_object_dict)):
             object_list_str.append(
             f"{predicted_label} - X:{int(x0)} Y: {int(y0)} Width: {int(width)} Height: {int(height)}"
             )
@@ -145,6 +145,6 @@ with gr.Blocks() as demo:
         output_desc = gr.Textbox(label="Description for using in ChatGPT", lines=5)
         # outputjson = gr.JSON(label="Detected Objects")
 
-    btn_detic.click(fn=inference, inputs=inp, outputs=[outviz, output_desc])
+    btn_detic.click(fn=inference, inputs=[inp,"checkbox"], outputs=[outviz, output_desc])
 
 demo.launch()