nickmuchi commited on
Commit
802125f
1 Parent(s): 1f0eb14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -65,7 +65,7 @@ def detect_objects(model_name,url_input,image_input,threshold):
65
 
66
  model = YolosForObjectDetection.from_pretrained(model_name)
67
 
68
- if url_input and validators.url(url_input):
69
  image = Image.open(requests.get(url_input, stream=True).raw)
70
 
71
  elif image_input:
@@ -98,6 +98,8 @@ Links to HuggingFace Models:
98
  """
99
 
100
  models = ["facebook/detr-resnet-50","facebook/detr-resnet-101",'hustvl/yolos-small']
 
 
101
 
102
  css = '''
103
  h1#title {
@@ -115,14 +117,11 @@ with demo:
115
  with gr.Tabs():
116
  with gr.TabItem('Image URL'):
117
  with gr.Row():
118
- url_input = gr.Textbox(lines=1,label='Enter valid image URL here..')
119
  img_output_from_url = gr.Image(shape=(450,450))
120
 
121
  with gr.Row():
122
- urls = ["https://media-cldnry.s-nbcnews.com/image/upload/t_fit-1500w,f_auto,q_auto:best/newscms/2020_14/3290756/200331-wall-street-ew-#343p.jpg"]
123
- example_url = gr.Dataset(components=[url_input],
124
- samples=[[url]
125
- for url in urls])
126
 
127
  url_but = gr.Button('Detect')
128
 
 
65
 
66
  model = YolosForObjectDetection.from_pretrained(model_name)
67
 
68
+ if validators.url(url_input):
69
  image = Image.open(requests.get(url_input, stream=True).raw)
70
 
71
  elif image_input:
 
98
  """
99
 
100
  models = ["facebook/detr-resnet-50","facebook/detr-resnet-101",'hustvl/yolos-small']
101
+ urls = ["https://media-cldnry.s-nbcnews.com/image/upload/t_fit-1500w,f_auto,q_auto:best/newscms/2020_14/3290756/200331-wall-street-ew-#343p.jpg"]
102
+
103
 
104
  css = '''
105
  h1#title {
 
117
  with gr.Tabs():
118
  with gr.TabItem('Image URL'):
119
  with gr.Row():
120
+ url_input = gr.Textbox(lines=2,label='Enter valid image URL here..')
121
  img_output_from_url = gr.Image(shape=(450,450))
122
 
123
  with gr.Row():
124
+ example_url = gr.Dataset(components=[url_input],samples=[[str(url)] for url in urls])
 
 
 
125
 
126
  url_but = gr.Button('Detect')
127