Spaces:

michaelwja
/

burn-detection

Runtime error

michaelwja commited on May 29, 2023

Commit

c49c9b0

1 Parent(s): 08b7eef

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.pt', source='local', device="cpu")
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)

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