Update app.py

app.py

@@ -1,25 +1,22 @@
-import subprocess
-import tempfile
-import time
-from pathlib import Path
+import tensorflow as tf
 import cv2
+import numpy as np
+from glob import glob
+#from models import Yolov4
 import gradio as gr
-import torch
-from ultralytics.yolo.v8.detect.predict import predict
+#model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
+def gradio_wrapper(img):
+    %cd %cd /content/YOLOv8-DeepSORT-Object-Tracking-and-Pulse-Estimation-2/ultralytics/yolo/v8/detect
+    !python predict.py model=yolov8x6.pt source=img
+    #print(np.shape(img))
+    results = model.predict(img)
+    return results[0]
+demo = gr.Interface(
+    gradio_wrapper,
+    #gr.Image(source="webcam", streaming=True, flip=True),
+    gr.Image(source="webcam", streaming=True),
+    "image",
+    live=True
+)
 
-
-
-def fn_image(image):
-    !python predict.py model=yolov8x6.pt source=image
-    
-gr.Interface(
-    fn_image,
-    inputs=[
-        gr.Image(source='webcam', streaming=True),
-    ],
-    outputs=gr.Image(type="pil"),
-    live=True,
-    title="YOLOv8",
-    allow_flagging='never',
-    #allow_screenshot='never',
-)
+demo.launch()