Sa-m commited on
Commit
2fafbac
·
1 Parent(s): 4d31a6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def find(name, path):
21
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt', force_reload=True)
22
  def detect(inp):
23
  output=model(inp)
24
- outPath='./yolov5/runs/detect/exp/'+inp_vid.split('/')[2]
25
  #out_path='./yolov5/runs/detect/exp/'+inp_vid.split('/')[2]
26
  return outPath
27
 
 
21
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt', force_reload=True)
22
  def detect(inp):
23
  output=model(inp)
24
+ outPath='./yolov5/runs/detect/exp/'+inp.split('/')[2]
25
  #out_path='./yolov5/runs/detect/exp/'+inp_vid.split('/')[2]
26
  return outPath
27