Update app.py

app.py

@@ -2,6 +2,10 @@ import os, json, cv2, numpy as np, gradio as gr
 from dataclasses import dataclass
 from typing import List, Dict, Tuple
 
+# --- Quiet Ultralytics settings warning and ensure a writeable config dir ---
+os.environ.setdefault("YOLO_CONFIG_DIR", "/tmp/Ultralytics")
+os.makedirs("/tmp/Ultralytics", exist_ok=True)
+
 # ---------- Config ----------
 @dataclass
 class InspectConfig:
@@ -40,11 +44,11 @@ YOLO_MODEL_REPO = os.getenv("YOLO_MODEL_REPO", "")      # e.g. "vaibhavi18092002
 YOLO_MODEL_FILE = os.getenv("YOLO_MODEL_FILE", "best.pt")
 
 def _ensure_weights():
+    """Ensure YOLO_MODEL_PATH exists; if not, try to download from a HF model repo."""
     global YOLO_MODEL_PATH
     if os.path.exists(YOLO_MODEL_PATH):
         return
     if YOLO_MODEL_REPO:
-        # single-file download from a HF model repo
         try:
             from huggingface_hub import hf_hub_download
             YOLO_MODEL_PATH = hf_hub_download(
@@ -71,6 +75,8 @@ def detect_boxes(frame: np.ndarray, cfg: InspectConfig = CFG) -> List[Dict]:
         res = _get_yolo().predict(source=frame, verbose=False, conf=0.25)[0]
         boxes = []
         H, W = frame.shape[:2]
+        if res.boxes is None or len(res.boxes) == 0:
+            return boxes
         for x1, y1, x2, y2 in res.boxes.xyxy.cpu().numpy():
             x1, y1, x2, y2 = map(int, [x1, y1, x2, y2])
             x1, y1 = max(0, x1), max(0, y1)
@@ -309,8 +315,12 @@ with gr.Blocks(theme=theme, css=custom_css, fill_height=True) as demo:
             with gr.Column(scale=1, elem_classes=["section"]):
                 dl_img = gr.File(label="Download annotated image")
                 dl_json = gr.File(label="Download JSON report")
-        img_btn.click(ui_image, inputs=[img_in, size_tol, appearance_thr, solidity_thr, seal_frac],
-                      outputs=[img_out, json_out, dl_img, dl_json])
+        img_btn.click(
+            ui_image,
+            inputs=[img_in, size_tol, appearance_thr, solidity_thr, seal_frac],
+            outputs=[img_out, json_out, dl_img, dl_json],
+            concurrency_limit=2  # Gradio 4: per-event limit
+        )
     with gr.Tab("Video"):
         with gr.Row():
             with gr.Column(scale=1, elem_classes=["section"]):
@@ -324,8 +334,12 @@ with gr.Blocks(theme=theme, css=custom_css, fill_height=True) as demo:
             with gr.Column(scale=1, elem_classes=["section"]):
                 dl_vid = gr.File(label="Download annotated video")
                 dl_jsonl = gr.File(label="Download JSONL report")
-        vid_btn.click(ui_video, inputs=[vid_in, size_tol, appearance_thr, solidity_thr, seal_frac],
-                      outputs=[vid_out, vjson_out, dl_vid, dl_jsonl])
+        vid_btn.click(
+            ui_video,
+            inputs=[vid_in, size_tol, appearance_thr, solidity_thr, seal_frac],
+            outputs=[vid_out, vjson_out, dl_vid, dl_jsonl],
+            concurrency_limit=1  # keep video runs serialized
+        )
 
 # In Spaces: no share=True, no host/port tweaks needed.
-demo.queue(concurrency_count=2, max_size=32).launch()
+demo.queue(max_size=32).launch()