Spaces:

devisionx
/

Materials-Demo

Runtime error

mennamostafa55555 commited on Sep 6, 2023

Commit

e8d8961

1 Parent(s): c75d65a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def yolov8_inference(
 ):
     image=image[:, :, ::-1].astype(np.uint8)
-    model = YOLO("/content/best_weights.pt")
     results = model(image,imgsz=360)[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/Final_demo/blob/main/best_weights.pt")
     results = model(image,imgsz=360)[0]
     image=image[:, :, ::-1].astype(np.uint8)
     detections = sv.Detections.from_yolov8(results)