Update app.py

app.py

@@ -41,6 +41,13 @@ def visualize_one_hot_encoded_sequence(tensor, sentence, col_labels, cmap='BuGn'
     ax.set_xlabel('Phones')
     ax.set_ylabel('Features')
 
+    ax.set_xticks(np.arange(-0.5, tensor.shape[1], 1), minor=True)
+    ax.set_yticks(np.arange(-0.5, tensor.shape[0], 1), minor=True)
+    
+    # Turn on the grid for minor ticks (i.e., between the cells)
+    ax.grid(which='minor', color='darkgrey', linestyle='-', linewidth=2)
+    ax.tick_params(which='minor', size=0)  # Hide minor tick marks
+
     # Display the heatmap
     ax.set_title(f"»{sentence}«")
     return fig