Kaushik Bar commited on
Commit
efcdb4f
1 Parent(s): 16c0c2f
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -95,11 +95,10 @@ Links to HuggingFace Models:
95
  - [facebook/detr-resnet-101](https://huggingface.co/facebook/detr-resnet-101)
96
  - [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small)
97
  - [hustvl/yolos-tiny](https://huggingface.co/hustvl/yolos-tiny)
98
- - [mhyatt000/yolov5](https://huggingface.co/mhyatt000/YOLOv5)
99
 
100
  """
101
 
102
- models = ["facebook/detr-resnet-50","facebook/detr-resnet-101",'hustvl/yolos-small','hustvl/yolos-tiny', 'mhyatt000/yolov5']
103
  urls = ["https://c8.alamy.com/comp/J2AB4K/the-new-york-stock-exchange-on-the-wall-street-in-new-york-J2AB4K.jpg"]
104
 
105
  css = '''
@@ -113,7 +112,7 @@ with demo:
113
  gr.Markdown(title)
114
  gr.Markdown(description)
115
  options = gr.Dropdown(choices=models,label='Select Object Detection Model',show_label=True)
116
- slider_input = gr.Slider(minimum=0.2,maximum=1,value=0.7,label='Prediction Threshold')
117
 
118
  with gr.Tabs():
119
  with gr.TabItem('Image URL'):
 
95
  - [facebook/detr-resnet-101](https://huggingface.co/facebook/detr-resnet-101)
96
  - [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small)
97
  - [hustvl/yolos-tiny](https://huggingface.co/hustvl/yolos-tiny)
 
98
 
99
  """
100
 
101
+ models = ["facebook/detr-resnet-50","facebook/detr-resnet-101",'hustvl/yolos-small','hustvl/yolos-tiny']
102
  urls = ["https://c8.alamy.com/comp/J2AB4K/the-new-york-stock-exchange-on-the-wall-street-in-new-york-J2AB4K.jpg"]
103
 
104
  css = '''
 
112
  gr.Markdown(title)
113
  gr.Markdown(description)
114
  options = gr.Dropdown(choices=models,label='Select Object Detection Model',show_label=True)
115
+ slider_input = gr.Slider(minimum=0.2,maximum=1,value=0.5,label='Prediction Threshold')
116
 
117
  with gr.Tabs():
118
  with gr.TabItem('Image URL'):