Spaces:

rosebe
/

EcoSmart

Runtime error

rosebe commited on Jun 30, 2023

Commit

5558e72

1 Parent(s): 018a50d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import shutil
 # Load the model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
-model2 = path='best.pt
 def detect_image(image):
     results = model2(image)

 # Load the model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
+model2 = torch.hub.load('ultralytics/yolov5', 'yolov5s', path = 'best.pt')
 def detect_image(image):
     results = model2(image)