Update app.py

app.py

@@ -132,7 +132,7 @@ if sent:
     ax.spines[["right", "top"]].set_visible(False)
 
     # dialect_labels = [int(dialect in valid_dialects) for dialect in DIALECTS]
-    im = ax.imshow(probabilities, cmap="vanimo", alpha=0.5, vmin=0, vmax=1)
+    im = ax.imshow([probabilities], cmap="vanimo", alpha=0.5, vmin=0, vmax=1)
     ax.set_xticks(range(len(DIALECTS)))
     ax.set_xticklabels(DIALECTS, fontsize=8, rotation=90, ha="right")
     ax.set_yticks([])