Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 import os 
 import requests
 import json
+import cv2 
 
 from PIL import Image
 from huggingface_hub import hf_hub_download
@@ -103,9 +104,44 @@ def yolov8_img_inference(
     # output_image = visualize_object_predictions(image=image, object_prediction_list=object_prediction_list)
     # return output_image['image']
     # render = render_result(model=model, image=image, result=results[0])
- 
 
-        
+def yolov8_vid_inference(video_path):
+    cap = cv2.VideoCapture(video_path)
+    while cap.isOpened():
+        success, frame = cap.read()
+
+        if success:
+            frame_copy = frame.copy()
+            outputs = model.predict(source=frame)
+            results = outputs[0].cpu().numpy()
+            for i, det in enumerate(results.boxes.xyxy):
+                cv2.rectangle(
+                    frame_copy,
+                    (int(det[0]), int(det[1])),
+                    (int(det[2]), int(det[3])),
+                    color=(0, 0, 255),
+                    thickness=2,
+                    lineType=cv2.LINE_AA
+                )
+            yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
+    
+
+inputs_vid = [
+    gr.components.Video(type="filepath", label="Input Video"),
+]
+
+outputs_vid = [
+    gr.components.Image(type="numpy", label="Output Image"),
+              ]
+
+interface_vid = gr.Interface(
+    fn=yolov8_vid_inference,
+    inputs = inputs_vid,
+    outputs = outputs_vid,
+    title = app_title,
+    cache_examples = False,
+    
+)
 
 inputs_image = [
     # gr.inputs.Image(type="filepath", label="Input Image"),
@@ -141,9 +177,9 @@ interface_image = gr.Interface(
     
 )
 
-# gr.TabbedInterface(
-#     [interface_image],
-#     tab_names=['Image inference']
-# ).queue().launch()
+gr.TabbedInterface(
+    [interface_image, interface_vid],
+    tab_names=['Image inference', 'Video inference']
+).queue().launch()
 
-interface_image.launch(debug=True, enable_queue=True)
+# interface_image.launch(debug=True, enable_queue=True)