Update app.py

app.py

@@ -2,14 +2,13 @@ import gradio as gr
 from ultralytics import YOLO
 import cv2
 from deep_sort_realtime.deepsort_tracker import DeepSort
-import tempfile
 
 # Initialize YOLO model
 model = YOLO("yolov8l.pt")  # Load YOLOv8 model
 tracker = DeepSort(max_age=30, n_init=3, nn_budget=100)
 
-def count_people_in_video(video_file):
-    cap = cv2.VideoCapture(video_file)  # Load video
+def count_people_in_video(video_path):
+    cap = cv2.VideoCapture(video_path)  # Load video
     total_ids = set()  # Track unique IDs
 
     while cap.isOpened():
@@ -42,11 +41,9 @@ def count_people_in_video(video_file):
 
 # Gradio Interface
 def process_video(video_file):
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
-        temp_file.write(video_file.read())
-        temp_file.flush()
-        total_people = count_people_in_video(temp_file.name)
-        return f"Total unique people in the video: {total_people}"
+    # The `video_file` is a path to the temporary file
+    total_people = count_people_in_video(video_file)
+    return f"Total unique people in the video: {total_people}"
 
 interface = gr.Interface(
     fn=process_video,
@@ -57,4 +54,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)