latest changes

app.py

@@ -84,6 +84,7 @@ current_prediction_label = "Initializing..."
 current_probabilities = {label: 0.0 for label in CLASS_LABELS}
 
 # --- 4. Gradio Live Inference Function (Generator) ---
+# This function will receive individual frames from the webcam as a NumPy array (H, W, C, RGB)
 def predict_live_frames(input_frame):
     global frame_buffer, current_prediction_label, current_probabilities
 
@@ -137,41 +138,27 @@ def predict_live_frames(input_frame):
 
     yield cv2.cvtColor(display_frame, cv2.COLOR_BGR2RGB)
 
-# --- 5. Gradio Blocks Interface Setup ---
-with gr.Blocks(
-    title="Real-time Violence Detection", # Title for the browser tab
-    # REMOVED: theme=gr.themes.Default(primary_hue=gr.Color(...)) to fix the AttributeError
-) as demo:
-    gr.Markdown(
-        """
-        # 🎬 Real-time Violence Detection
-        **Live Feed with Constant Predictions**
-
-        This model analyzes your live webcam feed for violence, displaying the predicted class and probabilities on the screen.
-        Please grant webcam access when prompted by your browser.
-        """
-    )
+# --- 5. Gradio Interface Setup (with hidden buttons) ---
+iface = gr.Interface(
+    fn=predict_live_frames,
+    # Input: Live webcam feed, configure for streaming
+    inputs=gr.Video(sources=["webcam"], streaming=True, label="Live Webcam Feed"),
+    # Output: Image component to display processed frames
+    outputs=gr.Image(type="numpy", label="Processed Feed with Predictions"),
     
-    with gr.Row():
-        video_input = gr.Video(
-            sources=["webcam"], 
-            streaming=True, 
-            label="Live Webcam Feed",
-            height=480, 
-            width=640 
-        )
-        
-        video_output = gr.Image(
-            type="numpy", 
-            label="Processed Feed with Predictions",
-            height=480, 
-            width=640 
-        )
-
-    video_input.stream(
-        predict_live_frames, 
-        inputs=video_input,  
-        outputs=video_output 
-    )
+    title="Real-time Violence Detection with SmallVideoClassifier (Webcam)",
+    description=(
+        "This model analyzes your live webcam feed for violence, displaying the predicted class and probabilities on the screen. "
+        "Please grant webcam access when prompted by your browser."
+    ),
+    
+    # --- IMPORTANT: Hide the default submit/clear buttons ---
+    submit_btn=None,
+    clear_btn=None,
+    
+    allow_flagging="never",
+    # No examples needed for live webcam
+    examples=None 
+)
 
-demo.launch()
+iface.launch()