Update app.py

app.py

@@ -28,7 +28,7 @@ _MODEL = get_model()
 
 def plot(attentions: np.ndarray):
     """Plots the attention maps from individual attention heads."""
-    fig, axes = plt.subplots(nrows=1, ncols=4) # figsize=(13, 13)
+    fig, axes = plt.subplots(nrows=1, ncols=4, figsize=(13, 13)) 
     img_count = 0
 
     for i in range(attentions.shape[-1]):