Update app.py

app.py

@@ -28,7 +28,7 @@ def process_sequence(sequence, domain_bounds, n):
     all_logits = []
 
     for i in range(len(sequence)):
-        if start_index <= i <= end_index:
+        if start_index <= i <= (end_index - 1):
             masked_seq = sequence[:i] + '<mask>' + sequence[i+1:]
             inputs = tokenizer(masked_seq, return_tensors="pt", padding=True, truncation=True, max_length=2000)
             inputs = {k: v.to(device) for k, v in inputs.items()}
@@ -56,8 +56,7 @@ def process_sequence(sequence, domain_bounds, n):
     transposed_logits_array = normalized_logits_array.T
 
     # Plotting the heatmap
-    step = 50
-    y_tick_positions = np.arange(0, len(sequence), step)
+    y_tick_positions = np.arange((start_index+1), end_index + 1, 10)
     y_tick_labels = [str(pos) for pos in y_tick_positions]
     
     plt.figure(figsize=(15, 8))