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>""")
|