kisa-misa's picture
Update app.py
5437fbc
raw
history blame
534 Bytes
import tensorflow as tf
import cv2
import numpy as np
from glob import glob
#from models import Yolov4
import gradio as gr
#model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
def gradio_wrapper(img):
%cd ultralytics/yolo/v8/detect/
!python predict.py model=yolov8x6.pt source=img
#print(np.shape(img))
demo = gr.Interface(
gradio_wrapper,
#gr.Image(source="webcam", streaming=True, flip=True),
gr.Image(source="webcam", streaming=True),
"image",
live=True
)
demo.launch()