Update app.py

app.py

@@ -56,8 +56,8 @@ def process_sequence(sequence, domain_bounds, n):
     normalized_logits_array = F.softmax(torch.tensor(all_logits_array), dim=-1).numpy()
     transposed_logits_array = normalized_logits_array.T
 
-    # Plotting the heatmap
-    x_tick_positions = np.arange(start_index, end_index)
+   # Plotting the heatmap
+    x_tick_positions = np.arange(start_index, end_index, 10)
     x_tick_labels = [str(pos + 1) for pos in x_tick_positions]
     
     plt.figure(figsize=(15, 8))
@@ -66,7 +66,7 @@ def process_sequence(sequence, domain_bounds, n):
     plt.ylabel('Token')
     plt.xlabel('Residue Index')
     plt.yticks(rotation=0)
-    plt.xticks(x_tick_positions, x_tick_labels, rotation = 0)
+    plt.xticks(x_tick_positions - start_index, x_tick_labels, rotation=0) 
     
     # Save the figure to a BytesIO object
     buf = BytesIO()