Update app.py

app.py

@@ -2,13 +2,12 @@ import tensorflow as tf
 import cv2
 import numpy as np
 from glob import glob
-from from ultralytics.yolo.v8.detect import predict
+from ultralytics.yolo.v8.detect import predict
 #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
+    predict(model="yolov8x6.pt", source=img)
     #print(np.shape(img))
 demo = gr.Interface(
     gradio_wrapper,