ClassCat commited on
Commit
4f82037
1 Parent(s): d9f6241

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,7 +77,7 @@ with gr.Blocks(title="DETR Object Detection - ClassCat",
77
 
78
  gr.HTML("""<h4 style="color:navy;">1. Select a model.</h4>""")
79
 
80
- model = gr.Radio(["detr-resnet-50", "detr-resnet-101"], value="detr-resnet-50", label="Select a model")
81
 
82
  gr.HTML("""<br/>""")
83
  gr.HTML("""<h4 style="color:navy;">2-a. Select an example by clicking a thumbnail below.</h4>""")
@@ -90,7 +90,7 @@ with gr.Blocks(title="DETR Object Detection - ClassCat",
90
  gr.Examples(['samples/cats.jpg', 'samples/detectron2.png', 'samples/cat.jpg', 'samples/hotdog.jpg'], inputs=input_image)
91
 
92
  gr.HTML("""<br/>""")
93
- gr.HTML("""<h4 style="color:navy;">3. Then, click "Infer" button to predict object instances. It will take about 15-20 seconds (on cpu)</h4>""")
94
 
95
  send_btn = gr.Button("Infer")
96
  send_btn.click(fn=infer, inputs=[model, input_image], outputs=[output_image])
 
77
 
78
  gr.HTML("""<h4 style="color:navy;">1. Select a model.</h4>""")
79
 
80
+ model = gr.Radio(["detr-resnet-50", "detr-resnet-101"], value="detr-resnet-50", label="Model name")
81
 
82
  gr.HTML("""<br/>""")
83
  gr.HTML("""<h4 style="color:navy;">2-a. Select an example by clicking a thumbnail below.</h4>""")
 
90
  gr.Examples(['samples/cats.jpg', 'samples/detectron2.png', 'samples/cat.jpg', 'samples/hotdog.jpg'], inputs=input_image)
91
 
92
  gr.HTML("""<br/>""")
93
+ gr.HTML("""<h4 style="color:navy;">3. Then, click "Infer" button to predict object instances. It will take about 10 seconds (on cpu)</h4>""")
94
 
95
  send_btn = gr.Button("Infer")
96
  send_btn.click(fn=infer, inputs=[model, input_image], outputs=[output_image])