Upload app.py

app.py

@@ -45,8 +45,6 @@ def predict_video(video):
     temp_output_path = tempfile.mktemp(suffix=".mp4")
     fourcc = cv2.VideoWriter_fourcc(*'mp4v') 
     out = cv2.VideoWriter(temp_output_path, fourcc, fps, (frame_width, frame_height))
-    
-    processed_frames = []
 
     while True:
         ret, frame = cap.read()
@@ -55,18 +53,20 @@ def predict_video(video):
         
         result = detection_model.predict(frame, conf=0.5, iou=0.6)
         img_bgr = result[0].plot()
-        out_frame = img_bgr[..., ::-1] 
-        processed_frames.append(Image.fromarray(out_frame))  # Store frame as PIL image
+        out_frame = img_bgr[..., ::-1]  # Convert BGR to RGB
         out.write(out_frame)
     
     cap.release()
     out.release()
+    
+    # Return the path to the processed video
     return temp_output_path  # Return the path to the processed video
 
 def enable_button(image_input, video_input):
     if image_input is None and video_input is None:
         return gr.Button.update(interactive=False) 
-    return gr.Button.update(interactive=True) 
+    return gr.Button.update(interactive=True)
+
 
 
 REPO_ID = "dexpyw/model"
@@ -85,7 +85,7 @@ image_interface = gr.Interface(
 video_interface = gr.Interface(
     fn=predict_video,
     inputs=gr.Video(label="Upload Video"),
-    outputs=gr.Gallery(label="Predicted Video Frames"),
+    outputs=gr.Video(label="Predicted Video"),
     live=False
 )