josedolot commited on
Commit
2698e40
·
1 Parent(s): 7405ff8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ def inference(img):
12
  #os.system('python hybridnets_test.py -w weights/hybridnets.pth --source demo/image --output demo_result --imshow False --imwrite True --cuda False')
13
  os.system('python hybridnets_test.py -w weights/hybridnets.pth --source demo/image --output demo_result --imshow False --imwrite True --cuda False --float16 False')
14
 
15
- return 'demo_result/1.jpg'
16
 
17
  title="HybridNets Demo"
18
 
@@ -21,7 +21,7 @@ description="Gradio demo for HybridNets: End2End Perception Network pretrained o
21
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2203.09035' target='_blank'>ybridNets: End2End Perception Network</a> | <a href='https://github.com/datvuthanh/HybridNets' target='_blank'>Github Repo</a></p>"
22
 
23
 
24
- examples=[['1.jpg']]
25
  gr.Interface(
26
  inference,
27
  [gr.inputs.Image(type="pil", label="Input")],
 
12
  #os.system('python hybridnets_test.py -w weights/hybridnets.pth --source demo/image --output demo_result --imshow False --imwrite True --cuda False')
13
  os.system('python hybridnets_test.py -w weights/hybridnets.pth --source demo/image --output demo_result --imshow False --imwrite True --cuda False --float16 False')
14
 
15
+ return 'demo_result/0.jpg'
16
 
17
  title="HybridNets Demo"
18
 
 
21
  article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2203.09035' target='_blank'>ybridNets: End2End Perception Network</a> | <a href='https://github.com/datvuthanh/HybridNets' target='_blank'>Github Repo</a></p>"
22
 
23
 
24
+ examples=[['1.jpg','2.jpg']]
25
  gr.Interface(
26
  inference,
27
  [gr.inputs.Image(type="pil", label="Input")],