Update app.py

app.py

@@ -49,7 +49,7 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
     while loop:
         end = start  + window_length
         # If the end index exceeds total length, set the flag to False and adjust the end index
-        if end >= total_len:
+        if end >= total_len OR count == 25:
             loop = False
             end = total_len