Spaces:

michaelwja
/

burn-detection

Runtime error

michaelwja commited on May 29, 2023

Commit

08b7eef

•

1 Parent(s): b01fc15

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def yolov7_inference(
         Rendered image
     """
-    model = torch.hub.load('WongKinYiu/yolov7', 'custom', path='skin_burn_2022_8_21 (1).pt', source='local', device="cpu", hf_model=True, trace=False)
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)

         Rendered image
     """
+    model = torch.hub.load('WongKinYiu/yolov7', 'custom', path='skin_burn.pt', source='local', device="cpu")
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)