Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -63,9 +63,12 @@ def detect_objects(model_name,url,image_upload,threshold):
|
|
63 |
elif 'yolos' in model_name:
|
64 |
|
65 |
model = YolosForObjectDetection.from_pretrained(model_name)
|
|
|
|
|
66 |
|
67 |
-
|
68 |
-
|
|
|
69 |
elif image_upload:
|
70 |
image = image_upload
|
71 |
|
@@ -105,14 +108,14 @@ with demo:
|
|
105 |
slider_input = gr.Slider(minimum=0.2,maximum=1,value=0.7,label='Prediction Threshold')
|
106 |
|
107 |
with gr.Tabs():
|
108 |
-
with gr.
|
109 |
with gr.Row():
|
110 |
url_input = gr.Textbox(lines=1,label='Enter valid image URL here..')
|
111 |
img_output_from_url = gr.Image(shape=(450,450))
|
112 |
|
113 |
url_but = gr.Button('Detect')
|
114 |
|
115 |
-
with gr.
|
116 |
with gr.Row():
|
117 |
img_input = gr.Image(type='pil')
|
118 |
img_output_from_upload= gr.Image(shape=(450,450))
|
|
|
63 |
elif 'yolos' in model_name:
|
64 |
|
65 |
model = YolosForObjectDetection.from_pretrained(model_name)
|
66 |
+
|
67 |
+
if url:
|
68 |
|
69 |
+
if validators.url(url):
|
70 |
+
image = Image.open(requests.get(url, stream=True).raw)
|
71 |
+
|
72 |
elif image_upload:
|
73 |
image = image_upload
|
74 |
|
|
|
108 |
slider_input = gr.Slider(minimum=0.2,maximum=1,value=0.7,label='Prediction Threshold')
|
109 |
|
110 |
with gr.Tabs():
|
111 |
+
with gr.TabItem('Image URL'):
|
112 |
with gr.Row():
|
113 |
url_input = gr.Textbox(lines=1,label='Enter valid image URL here..')
|
114 |
img_output_from_url = gr.Image(shape=(450,450))
|
115 |
|
116 |
url_but = gr.Button('Detect')
|
117 |
|
118 |
+
with gr.TabItem('Image Upload'):
|
119 |
with gr.Row():
|
120 |
img_input = gr.Image(type='pil')
|
121 |
img_output_from_upload= gr.Image(shape=(450,450))
|