Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -65,10 +65,13 @@ def get_results(image):
|
|
65 |
|
66 |
return plt, {labels[i]: float(prediction[i]) for i in range(10)}
|
67 |
|
|
|
|
|
68 |
iface = gr.Interface(
|
69 |
fn=get_results,
|
70 |
title = "Patch ConvNet Demo",
|
71 |
description = "This space is a demo of the paper 'Augmenting Convolutional networks with attention-based aggregation' π",
|
|
|
72 |
inputs=gr.inputs.Image(label="Input Image"),
|
73 |
outputs=[
|
74 |
gr.outputs.Image(label="Attention Map"),
|
|
|
65 |
|
66 |
return plt, {labels[i]: float(prediction[i]) for i in range(10)}
|
67 |
|
68 |
+
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.13692' target='_blank'>Augmenting Convolutional networks with attention-based aggregation</a></p> <center>Contributors: <a href='https://twitter.com/ariG23498'>Aritra Roy Gosthipaty</a>|<a href='https://twitter.com/ritwik_raha'>Ritwik Raha</a>|<a href='https://twitter.com/Cr0wley_zz'>Devjyoti Chakraborty</a></center>"
|
69 |
+
|
70 |
iface = gr.Interface(
|
71 |
fn=get_results,
|
72 |
title = "Patch ConvNet Demo",
|
73 |
description = "This space is a demo of the paper 'Augmenting Convolutional networks with attention-based aggregation' π",
|
74 |
+
article = article,
|
75 |
inputs=gr.inputs.Image(label="Input Image"),
|
76 |
outputs=[
|
77 |
gr.outputs.Image(label="Attention Map"),
|