Update app.py

app.py

@@ -28,19 +28,17 @@ annotatormask=sv.MaskAnnotator()
 
 def yolov8_inference(
     image: gr.inputs.Image = None,
-    model_name: gr.inputs.Dropdown = None,
-    image_size: gr.inputs.Slider = 640,
-    conf_threshold: gr.inputs.Slider = 0.25,
+    conf_threshold: gr.inputs.Slider = 0.5,
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
 
-
-    model = YOLO("https://huggingface.co/spaces/devisionx/Sixth_Demo/blob/main/bestt_weight.pt")
-
-    results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=640)[0]
+    image=image[:, :, ::-1].astype(np.uint8)
+    model = YOLO("/content/segment/train/weights/best.pt")
+    results = model(image,imgsz=640)[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)
+    annotated_image = annotatormask.annotate(scene=image, detections=detections)
+    annotated_image = annotatorbbox.annotate(scene=annotated_image , detections=detections)