Update app.py

app.py

@@ -24,19 +24,19 @@ _MODEL = get_model()
 
 
 def plot(attentions: np.ndarray):
-	"""Plots the attention maps from individual attention heads."""
+    """Plots the attention maps from individual attention heads."""
     fig, axes = plt.subplots(nrows=1, ncols=4, figsize=(13, 13))
-	img_count = 0
+    img_count = 0
 
-	for i in range(attentions.shape[-1]):
-	    if img_count < attentions.shape[-1]:
-	        axes[i].imshow(attentions[:, :, img_count])
-	        axes[i].title.set_text(f"Attention head: {img_count}")
-	        axes[i].axis("off")
-	        img_count += 1
+    for i in range(attentions.shape[-1]):
+        if img_count < attentions.shape[-1]:
+            axes[i].imshow(attentions[:, :, img_count])
+            axes[i].title.set_text(f"Attention head: {img_count}")
+            axes[i].axis("off")
+            img_count += 1
 
-	fig.tight_layout()
-	return fig
+    fig.tight_layout()
+    return fig
 
 
 def show_plot(image):