Update app.py

app.py

@@ -44,6 +44,8 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
     
     loop = True
     count = 1
+    print(f'Total Length: {total_len}')
+    
     while loop:
         end = start  + window_length
         # If the end index exceeds total length, set the flag to False and adjust the end index
@@ -51,6 +53,9 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
             loop = False
             end = total_len
 
+        print(f'Start: {start}')
+        print(f'End: {end}')
+    
         # 1 => Define the text chunk
         input_ids_chunk = input_ids[start : end]
         attention_mask_chunk = attention_mask[start : end]