Update app.py

app.py

@@ -2,6 +2,7 @@ import supervision as sv
 import gradio as gr
 from ultralytics import YOLO
 import sahi
+import numpy as np
 
 
 
@@ -20,6 +21,8 @@ sahi.utils.file.download_from_url(
 )
 
 
+
+
 annotatorbbox = sv.BoxAnnotator()
 annotatormask=sv.MaskAnnotator()
 
@@ -33,13 +36,14 @@ def yolov8_inference(
 ):
 
 
-    model = YOLO("https://huggingface.co/spaces/devisionx/Fifth_model/blob/main/best_weigh.pt")
+   
+    image=image[:, :, ::-1].astype(np.uint8)
+    model = YOLO("/content/segment/train/weights/best.pt")
 
     results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=360)[0]
+    image=image[:, :, ::-1].astype(np.uint8)
     detections = sv.Detections.from_yolov8(results)
     annotated_image = annotatorbbox.annotate(scene=image, detections=detections)
-    annotated_image = annotatormask.annotate(scene=annotated_image, detections=detections)
-
 
 
 
@@ -71,4 +75,4 @@ demo_app = gr.Interface(examples=examples,
     cache_examples=True,
     theme="default",
 )
-demo_app.launch(debug=True, enable_queue=True)
+demo_app.launch(debug=False, enable_queue=True)