Update app.py

app.py

@@ -48,24 +48,24 @@ def process_sequence(sequence, domain_bounds, n):
 
     # checking domain bounds inputs
     try:
-      start = int(domain_bounds['start'][0]) 
+      start = int(domain_bounds['start'][0])
       end = int(domain_bounds['end'][0])
     except ValueError:
       raise gr.Error("Error: Start and end indices must be integers.")
       return None, None, None
     if start >= end:
-      raise gr.Error("Start index must be smaller than end index.") 
+      raise gr.Error("Start index must be smaller than end index.")
       return None, None, None
     if start == 0 and end != 0:
-      raise gr.Error("Indexing starts at 1. Please enter valid domain bounds.") 
+      raise gr.Error("Indexing starts at 1. Please enter valid domain bounds.")
       return None, None, None
-    if start == 0 or end == 0:
-      raise gr.Error("Domain bounds cannot be zero. Please enter valid domain bounds.") 
+    if start <= 0 or end <= 0:
+      raise gr.Error("Domain bounds must be positive integers. Please enter valid domain bounds.")
       return None, None, None
     if start > len(sequence) or end > len(sequence):
       raise gr.Error("Domain bounds exceed sequence length.")
       return None, None, None
-    
+
     # checking n inputs
     if n == None:
       raise gr.Error("Choose Top N Tokens from the dropdown menu.")
@@ -117,10 +117,12 @@ 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
+    # Plotting the heatmap
     domain_len = end - start
-    if domain_len > 100:
+    if 500 > domain_len > 100:
       step_size = 50
+    elif 500 <= domain_len:
+      step_size = 100
     elif domain_len < 10:
       step_size = 1
     else: