upp

app.py

@@ -1,3 +1,9 @@
+import os
+
+# Fix Ultralytics permission warning
+os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
+
+
 import gradio as gr
 from ultralytics import YOLO
 import cv2
@@ -15,36 +21,63 @@ def detect(image):
     if image is None:
         return None
 
+
+    # Run YOLO inference
     results = model(
         image,
         verbose=False
     )
 
+
+    # Draw detections
     annotated = results[0].plot()
 
+
+    # Convert BGR -> RGB
     annotated = cv2.cvtColor(
         annotated,
         cv2.COLOR_BGR2RGB
     )
 
+
     return annotated
 
 
+
 demo = gr.Interface(
+
     fn=detect,
+
     inputs=gr.Image(
         sources=["webcam"],
-        type="numpy"
+        type="numpy",
+        label="Mobile Camera"
     ),
+
     outputs=gr.Image(
-        type="numpy"
+        type="numpy",
+        label="AI Detection"
     ),
-    title="YOLO Webcam Demo"
+
+    title="AI Shop Security Demo",
+
+    description="""
+    Real-time shop monitoring prototype.
+
+    Model:
+    - YOLOv8 Nano
+
+    Detects:
+    - People
+    - Bags
+    - Objects
+
+    Demo only: detections should be reviewed by humans.
+    """
 )
 
 
-print("Launching Gradio...")
 
 demo.launch(
-    server_name="0.0.0.0",
+    server_name="0.0.0.0"
 )

requirements.txt

@@ -1,3 +1,4 @@
 ultralytics
 opencv-python-headless
 numpy
+spaces