Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
497e471
1
Parent(s):
7d39495
Update app.py
Browse files
app.py
CHANGED
@@ -12,6 +12,8 @@ from matplotlib import patches, lines
|
|
12 |
from matplotlib.patches import Polygon
|
13 |
import gradio as gr
|
14 |
|
|
|
|
|
15 |
torch.set_grad_enabled(False);
|
16 |
# standard PyTorch mean-std input image normalization
|
17 |
transform = T.Compose([
|
@@ -134,10 +136,10 @@ def plot_inference(im, caption):
|
|
134 |
|
135 |
|
136 |
|
137 |
-
title = "
|
138 |
-
description = "demo for
|
139 |
-
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2104.
|
140 |
-
|
141 |
gr.Interface(
|
142 |
plot_inference,
|
143 |
[gr.inputs.Image(type="pil", label="Input"), gr.inputs.Textbox(label="input text")],
|
@@ -145,4 +147,5 @@ gr.Interface(
|
|
145 |
title=title,
|
146 |
description=description,
|
147 |
article=article,
|
|
|
148 |
).launch(debug=True)
|
|
|
12 |
from matplotlib.patches import Polygon
|
13 |
import gradio as gr
|
14 |
|
15 |
+
torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/05/28/08/32/elephant-1421167_1280.jpg', 'elephant.jpg')
|
16 |
+
|
17 |
torch.set_grad_enabled(False);
|
18 |
# standard PyTorch mean-std input image normalization
|
19 |
transform = T.Compose([
|
|
|
136 |
|
137 |
|
138 |
|
139 |
+
title = "MDETR"
|
140 |
+
description = "Gradio demo for MDETR: Modulated Detection for End-to-End Multi-Modal Understanding. To use it, simply upload your image and add text, or click one of the examples to load them. Read more at the links below."
|
141 |
+
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2104.12763'>MDETR: Modulated Detection for End-to-End Multi-Modal Understanding</a> | <a href='https://github.com/ashkamath/mdetr'>Github Repo</a></p>"
|
142 |
+
examples =['elephant.jpg','an elephant']
|
143 |
gr.Interface(
|
144 |
plot_inference,
|
145 |
[gr.inputs.Image(type="pil", label="Input"), gr.inputs.Textbox(label="input text")],
|
|
|
147 |
title=title,
|
148 |
description=description,
|
149 |
article=article,
|
150 |
+
examples=examples
|
151 |
).launch(debug=True)
|