import subprocess
import tempfile
import time
from pathlib import Path
import cv2
import gradio as gr
import torch
import ultralytics.yolo.v8.detect.predict

pipeline = predict()

def fn_image(image):
    return pipeline(image)
    
gr.Interface(
    fn_image,
    inputs=[
        gr.Image(source='webcam', streaming=True),
    ],
    outputs=gr.Image(type="file"),
    live=True,
    title="YOLOv8",
    allow_flagging=False,
    allow_screenshot=False,
)