Update app.py

app.py

@@ -1,9 +1,11 @@
 from ultralytics import YOLO
+from PIL import Image
 import gradio as gr
 from huggingface_hub import snapshot_download
-import cv2
 import os
 
+model_path = "best_int8_openvino_model"
+
 def load_model(repo_id):
     download_dir = snapshot_download(repo_id)
     print(download_dir)
@@ -13,23 +15,22 @@ def load_model(repo_id):
     return detection_model
 
 
-def detect_image(upload_image):
-    if upload_image is None:
-        return None
-    source = upload_image
-    result = detection_model.predict(source)
-    out_upload_image = result[0].plot()
-    return out_upload_image
+def predict(pilimg):
+
+    source = pilimg
+    # x = np.asarray(pilimg)
+    # print(x.shape)
+    result = detection_model.predict(source, conf=0.5, iou=0.6)
+    img_bgr = result[0].plot()
+    out_pilimg = Image.fromarray(img_bgr[..., ::-1])  # RGB-order PIL image
+    
+    return out_pilimg
 
-def clear_image():
-        return None
 
 REPO_ID = "CTYiHui/EggDefect"
 detection_model = load_model(REPO_ID)
 
-iface = gr.Interface(fn=predict, 
-                     inputs=gr.Image(type="pil"),
-                     outputs="text",
-                     live=True)
-
-iface.launch()
+gr.Interface(fn=predict,
+             inputs=gr.Image(type="pil"),
+             outputs=gr.Image(type="pil")
+             ).launch(share=True)