ariG23498 HF staff commited on
Commit
8cc2b14
β€’
1 Parent(s): 76d5cab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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"),