Update app.py

app.py

@@ -94,7 +94,7 @@ def run_dfine_classify(image, encoder_choice, refs_path):
     refs = Path(refs_path.strip()) if refs_path and refs_path.strip() else Path(REFS_DIR)
     if not refs.is_dir():
         return None, f"Refs folder not found: {refs}"
-    # Tuned on COCO GT: conf=0.5, gap=0.02. Lower det_threshold/min_side so D-FINE picks up more objects (gun, phone, etc.) like local.
+
     out_img, text = run_single_image(
         image,
         refs_dir=refs,
@@ -168,9 +168,9 @@ with gr.Blocks(title="Small Object Detection") as app:
 
         with gr.TabItem("D-FINE + Classify"):
             gr.Markdown(
-                "**D-FINE** runs first (person/car grouping), then small-object crops are classified. "
+                "**D-FINE** runs first (person/car grouping), then small-object detections inside each group crop are classified. "
                 "Choose **Jina** or **Nomic** for the embedding/classification model. "
-                "Uses the **refs** folder (one subfolder per class, e.g. refs/phone/, refs/cigarette/) with reference images."
+                "Only **known classes** are drawn. **Unknown** detections are skipped."
             )
             with gr.Row():
                 with gr.Column(scale=1):
@@ -187,8 +187,8 @@ with gr.Blocks(title="Small Object Detection") as app:
                     )
                     btn_dfine = gr.Button("Run D-FINE + Classify", variant="primary")
                 with gr.Column(scale=1):
-                    out_img_dfine = gr.Image(label="Output (crops with labels)", height=IMG_HEIGHT)
-                    out_text_dfine = gr.Textbox(label="Crop predictions", lines=10, interactive=False)
+                    out_img_dfine = gr.Image(label="Output group crops", height=IMG_HEIGHT)
+                    out_text_dfine = gr.Textbox(label="Known-class predictions", lines=10, interactive=False)
             btn_dfine.click(
                 fn=run_dfine_classify,
                 inputs=[inp_dfine, encoder_choice, refs_path],
@@ -199,4 +199,4 @@ with gr.Blocks(title="Small Object Detection") as app:
 app.launch(
     server_name=os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"),
     server_port=int(os.environ.get("PORT", os.environ.get("GRADIO_SERVER_PORT", 7860))),
-)
+)