Spaces:

Nada2001
/

yoloX-rayPanoramicAnalysisStreamlit

Runtime error

Nada2001 commited on Apr 16, 2023

Commit

0b966d0

1 Parent(s): f59b834

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ 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)
   return results

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