Spaces:
Running
Running
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])
|