Fix begin notes

app.py

@@ -82,13 +82,14 @@ def process_relevances(input_tokens, all_relevances, generated_tokens):
                 current_width = 0
         return best_width, best_patch_start
 
-    output_with_notes = [(el, None) for el in generated_tokens[:kernel_width]]
-    for row in range(kernel_width, len(generated_tokens)):
-        best_width, best_patch_start = find_largest_contiguous_patch(significant_areas[row-kernel_width+1])
+    output_with_notes = []
+    for row in range(len(generated_tokens)-kernel_width+1):
+        best_width, best_patch_start = find_largest_contiguous_patch(significant_areas[row])
         if best_width is not None:
             output_with_notes.append((generated_tokens[row], (best_width, best_patch_start)))
         else:
             output_with_notes.append((generated_tokens[row], None))
+    output_with_notes += [(el, None) for el in generated_tokens[-kernel_width+1:]]
 
     # Fuse the notes for consecutive output tokens if necessary
     for i in range(len(output_with_notes)):