Spaces:

Nada2001
/

yoloX-rayPanoramicAnalysisStreamlit

Runtime error

Nada2001 commited on Apr 15, 2023

Commit

59c392f

1 Parent(s): d074af2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 import os
 def getResult(filename):
-  model_name='/content/yolov5/runs/train/exp/weights/best.pt'
   model = torch.hub.load( "/content/yolov5/",'custom', source='local', path = model_name, force_reload = True)
   img = "/content/"+filename+""
   results = model(img)

 import os
 def getResult(filename):
+  model_name='best.pt'
   model = torch.hub.load( "/content/yolov5/",'custom', source='local', path = model_name, force_reload = True)
   img = "/content/"+filename+""
   results = model(img)