Update app.py

app.py

@@ -63,6 +63,8 @@ def process_sequence(sequence, domain_bounds, n):
     x_tick_labels = [str(pos + 1) for pos in x_tick_positions]
     
     plt.figure(figsize=(15, 8))
+    plt.rcParams.update({'font.size': 16})
+
     sns.heatmap(transposed_logits_array, cmap='plasma', xticklabels=x_tick_labels, yticklabels=filtered_tokens)
     plt.title('Token Probability Heatmap')
     plt.ylabel('Token')
@@ -72,7 +74,7 @@ def process_sequence(sequence, domain_bounds, n):
     
     # Save the figure to a BytesIO object
     buf = BytesIO()
-    plt.savefig(buf, format='png')
+    plt.savefig(buf, format='png', dpi = 300)
     buf.seek(0)
     plt.close()