josedolot commited on
Commit
17ea543
·
1 Parent(s): 2bd4518

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ os.system('curl -L -o weights/hybridnets.pth https://github.com/datvuthanh/Hybri
9
  def inference(img):
10
 
11
  img.save("demo/image/1.jpg", "JPEG")
12
- os.system('python hybridnets_test.py -w weights/hybridnets.pth --source demo/image --output demo_result --imshow False --imwrite True')
13
 
14
  return 'demo_result/1.jpg'
15
 
 
9
  def inference(img):
10
 
11
  img.save("demo/image/1.jpg", "JPEG")
12
+ os.system('python hybridnets_test.py -w weights/hybridnets.pth --source demo/image --output demo_result --imshow False --imwrite True --cuda False')
13
 
14
  return 'demo_result/1.jpg'
15