Update app.py

app.py

@@ -78,6 +78,7 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
         print("########:", decoded , ":##############")
         
         probabilities = torch.nn.functional.softmax(outputs[0], dim = -1)
+        print("########:", probabilities , ":##############")
         proba_list.append(probabilities)
         start = end