update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def get_figure(in_pil_img, in_results):
|
|
| 53 |
def infer(model, in_pil_img):
|
| 54 |
|
| 55 |
results = None
|
| 56 |
-
if model == "detr-resnet-101"
|
| 57 |
results = detector101(in_pil_img)
|
| 58 |
else:
|
| 59 |
results = detector50(in_pil_img)
|
|
@@ -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")
|
| 81 |
|
| 82 |
gr.HTML("""<br/>""")
|
| 83 |
gr.HTML("""<h4 style="color:navy;">2-a. Select an example by clicking a thumbnail below.</h4>""")
|
|
|
|
| 53 |
def infer(model, in_pil_img):
|
| 54 |
|
| 55 |
results = None
|
| 56 |
+
if model == "detr-resnet-101":
|
| 57 |
results = detector101(in_pil_img)
|
| 58 |
else:
|
| 59 |
results = detector50(in_pil_img)
|
|
|
|
| 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>""")
|