Spaces:

Sa-m
/

YoloV5-Party-Symbol-Detector-V1

Runtime error

Sa-m commited on Feb 14, 2022

Commit

61a1dcc

1 Parent(s): 674051d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def find(name, path):
   if name in files:
    return os.path.join(root, name)
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
 model.conf = 0.45
 #, force_reload=True
 def detect(inp):

   if name in files:
    return os.path.join(root, name)
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt',device='gpu')
 model.conf = 0.45
 #, force_reload=True
 def detect(inp):