Update app.py

app.py

@@ -70,7 +70,7 @@ examples =[['cats.jpg']]
 
 iface = gr.Interface(fn=visualize_attention, 
                      inputs=gr.inputs.Image(shape=(480, 480), type="pil"), 
-                     outputs=[gr.outputs.Image(type='file', label=f'attention_head_{i}') for i in range(6)],
+                     outputs=[gr.outputs.Image(type='file', label=f'attention_head_{i}') for i in range(12)],
                      title=title,
                      description=description,
                      article=article,