Spaces:

devisionx
/

Fire-Smoke-Demo

Runtime error

mennamostafa55555 commited on Sep 6, 2023

Commit

a67a4e8

1 Parent(s): 59d30dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def yolov8_inference(
 ):
     image=image[:, :, ::-1].astype(np.uint8)
-    model = YOLO("/content/segment/train/weights/best.pt")
     results = model(image,imgsz=320)[0]
     image=image[:, :, ::-1].astype(np.uint8)
     detections = sv.Detections.from_yolov8(results)

 ):
     image=image[:, :, ::-1].astype(np.uint8)
+    model = YOLO("https://huggingface.co/spaces/devisionx/Second_demo/blob/main/best.pt")
     results = model(image,imgsz=320)[0]
     image=image[:, :, ::-1].astype(np.uint8)
     detections = sv.Detections.from_yolov8(results)