Update app.py

app.py

@@ -5,12 +5,12 @@ from pathlib import Path
 import cv2
 import gradio as gr
 import torch
-import ultralytics.yolo.v8.detect.predict
+from ultralytics.yolo.v8.detect.predict import predict
+
 
-pipeline = predict()
 
 def fn_image(image):
-    return pipeline(image)
+    return predict(model=yolov8x6, source=image)
     
 gr.Interface(
     fn_image,