latest changes

app.py

@@ -80,6 +80,11 @@ transform = transforms.Compose([
 
 # --- 4. Gradio Live Inference Function (Generator) ---
 # This function will receive individual frames from the webcam
+# Initialize global state for the generator function (before the predict function)
+frame_buffer = [] # Buffer for collecting frames for model input
+current_prediction_label = "Initializing..."
+current_probabilities = {label: 0.0 for label in CLASS_LABELS} # Initial probabilities
+
 def predict_live_frames(input_frame):
     global frame_buffer, current_prediction_label, current_probabilities # Use global to maintain state across calls
 
@@ -112,8 +117,6 @@ def predict_live_frames(input_frame):
         
         # --- Sliding Window ---
         # Keep the last few frames to allow continuous predictions
-        # For example, if NUM_FRAMES is 8, and we want a new prediction every 2 frames,
-        # we slide the window by 2:
         slide_window_by = 1 # Predict every frame (most "real-time" feel but highest compute)
                            # Or: NUM_FRAMES // 2 (e.g., predict every 4 frames for NUM_FRAMES=8)
                            # Or: NUM_FRAMES (non-overlapping windows, less frequent updates)
@@ -151,16 +154,11 @@ def predict_live_frames(input_frame):
     yield cv2.cvtColor(display_frame, cv2.COLOR_BGR2RGB)
 
 
-# --- Initialize global state for the generator function ---
-frame_buffer = [] # Buffer for collecting frames for model input
-current_prediction_label = "Initializing..."
-current_probabilities = {label: 0.0 for label in CLASS_LABELS} # Initial probabilities
-
 # --- 5. Gradio Interface Setup ---
 iface = gr.Interface(
     fn=predict_live_frames,
-    # Use gr.Webcam for direct webcam input
-    inputs=gr.Webcam(streaming=True, label="Live Webcam Feed for Violence Detection"),
+    # CORRECTED: Use gr.Video with sources=["webcam"] for webcam input
+    inputs=gr.Video(sources=["webcam"], streaming=True, label="Live Webcam Feed for Violence Detection"),
     # Outputs are updated continuously by the generator
     outputs=gr.Image(type="numpy", label="Live Prediction Output"), # Using Image as output for continuous frames
     title="Real-time Violence Detection with SmallVideoClassifier (Webcam)",