Update app.py

app.py

@@ -43,7 +43,7 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
     window_length = 510
     
     loop = True
-
+    count = 1
     while loop:
         end = start  + window_length
         # If the end index exceeds total length, set the flag to False and adjust the end index
@@ -68,6 +68,8 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
         outputs = model(**input_dict)
 
         decoded = tokenizer.decode(input_ids_chunk)
+        print("Loop Count:" + count)
+        count = count + 1
         print("########:", decoded , ":##############")
         
         probabilities = torch.nn.functional.softmax(outputs[0], dim = -1)