Spaces:

rosebe
/

EcoSmart

Runtime error

rosebe commited on Jun 30, 2023

Commit

d65319e

1 Parent(s): 5558e72

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,7 @@ import os
 import shutil
 # 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)

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