Update app.py

app.py

@@ -6,8 +6,9 @@ from ultralytics.yolo.v8.detect import predict
 import gradio as gr
 #model = Yolov4(weight_path="yolov4.weights", class_name_path='coco_classes.txt')
 def gradio_wrapper(img):
-    predict(model="YOLOv8-real-time/ultralytics/yolo/v8/detect/yolov8x6.pt", source=img)
+    result = predict(model="YOLOv8-real-time/ultralytics/yolo/v8/detect/yolov8x6.pt", source=img)
     #print(np.shape(img))
+    return result
 demo = gr.Interface(
     gradio_wrapper,
     #gr.Image(source="webcam", streaming=True, flip=True),