update app.py
Browse files
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="
|
| 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
|
| 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])
|