Kaushik Bar
commited on
Commit
·
efcdb4f
1
Parent(s):
16c0c2f
zsod
Browse files
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'
|
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.
|
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'):
|