Spaces:

sparsh007
/

ConstructionImageAnnotation

Build error

sparsh007 commited on Sep 28, 2024

Commit

30eaedd

verified ·

1 Parent(s): 23bb58e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ shutil.rmtree(torch.hub._get_torch_home(), ignore_errors=True)
 # Load the YOLOv5 model from the uploaded best.pt file
 model = torch.hub.load(
-    'ultralytics/yolov5', 'custom', path='best.pt', trust_repo=True
 )
 # Global variables to store images and labels for carousel functionality

 # Load the YOLOv5 model from the uploaded best.pt file
 model = torch.hub.load(
+    'ultralytics/yolov5', 'custom', path='best-3.pt', trust_repo=True
 )
 # Global variables to store images and labels for carousel functionality