Spaces:
Build error
Build error
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() |