chore: attention only plot

app.py

@@ -57,4 +57,4 @@ def plot_attention(image):
 iface = gr.Interface(
 	fn=plot_attention,
 	inputs=[gr.inputs.Image(label="Input Image")],
-	outputs="Attention Map").launch()
+	outputs="image").launch()