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"),
|