ClassCat commited on
Commit
d9f6241
·
1 Parent(s): 463eb87

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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>""")