Update app.py

app.py

@@ -79,6 +79,7 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
         
         probabilities = torch.nn.functional.softmax(outputs[0], dim = -1)
         proba_list.append(probabilities)
+        start = end
     
     return proba_list