Update app.py

app.py

@@ -1,37 +1,44 @@
-import gradio as gr
-import os
 import cv2
 from ultralytics import YOLO
+import numpy as np
+import os
+import gradio as gr
 
-def object_detection(video):
-    
-    model=YOLO("best.pt")
-        
-    cap = cv2.VideoCapture(video)
-    
-    while True:
-        ret, frame = cap.read()
-    
-        if not ret:
-            break
-    
-        results = model(frame)
-        for result in results:
-            box=result.boxes
-            
-            x1, y1, x2, y2 = map(int, box.xyxy[0])
-            print(x1, y1, x2, y2)
-            cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
-    
-        return cv2.imshow("img", frame)
+def fonk(video_path):
+  
+  model=YOLO("best.pt")
+  cap=cv2.VideoCapture(video_path)
 
+  frame_width = int(cap.get(3)) 
+  frame_height = int(cap.get(4))
+  size = (frame_width, frame_height)
+  output_video = cv2.VideoWriter('filename.mp4',  
+                         cv2.VideoWriter_fourcc(*'XVID'), 
+                         10, size) 
 
-iface = gr.Interface(object_detection, 
-                    gr.Video(), 
-                    "playable_video", 
-                    examples=[
-                        os.path.join(os.path.abspath(''), 
-                                     "cow-video-cows-mooing-and-grazing-in-a-field.mp4")], 
-                    cache_examples=True)
+  
+  while True:
+    ret, frame= cap.read()
+
+    if ret!=True:
+      break
+
+    results= model(frame)
+    for result in results:
+      box=result.boxes
 
-iface.launch(share=True)
+      x1, y1, x2, y2 = map(int, box.xyxy[0])
+      print(x1, y1, x2, y2)
+      frame= cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
+      output_video.write(frame)
+      
+      return output_video
+
+demo = gr.Interface(fonk,
+                    inputs= gr.Video(),
+                    outputs=gr.Video(),
+                    examples=["cow-video-cows-mooing-and-grazing-in-a-field.mp4"],
+                    title= "cows",
+                    cache_examples=True)
+demo.launch()
+