Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -28,5 +28,5 @@ def predict():
|
|
28 |
|
29 |
|
30 |
if __name__ == "__main__":
|
31 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='
|
32 |
app.run(host="0.0.0.0", port=7860,debug =True) # debug=True causes Restarting with stat
|
|
|
28 |
|
29 |
|
30 |
if __name__ == "__main__":
|
31 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best2.pt', force_reload=True) # force_reload to recache
|
32 |
app.run(host="0.0.0.0", port=7860,debug =True) # debug=True causes Restarting with stat
|