Update app.py

app.py

@@ -1,9 +1,9 @@
 import cv2
 import numpy as np
 import datetime
-import matplotlib.pyplot as plt
 import gradio as gr
 
+# Ensure these files are available in the Hugging Face Space working directory
 net = cv2.dnn.readNet("yolov3.weights", "yolov3.cfg")
 layer_names = net.getLayerNames()
 output_layers = [layer_names[i - 1] for i in net.getUnconnectedOutLayers()]
@@ -43,11 +43,6 @@ def detect_objects(image):
     
     return [(boxes[i], class_ids[i], confidences[i]) for i in range(len(boxes)) if i in indexes]
 
-def capture_frame(frame):
-    current_time = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-    cv2.imwrite(f"captured_frame_{current_time}.jpg", frame)
-    print(f"Captured frame saved as 'captured_frame_{current_time}.jpg'")
-
 def process_image(image):
     image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
     detections = detect_objects(image)