Spaces:

devisionx
/

Brain-Tumor-Demo

Runtime error

mennamostafa55555 commited on Sep 4, 2023

Commit

6ac1fcf

1 Parent(s): 4d26a0d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def yolov8_inference(
     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)

     image=image[:, :, ::-1].astype(np.uint8)
+    model = YOLO("https://huggingface.co/spaces/devisionx/Fifth_demo/blob/main/best_weigh.pt")
     results = model(image,conf=conf_threshold,iou=iou_threshold ,imgsz=360)[0]
     image=image[:, :, ::-1].astype(np.uint8)