Update app.py

app.py

@@ -79,7 +79,8 @@ 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)
-        return proba_list
+    
+    return proba_list
 
 def get_mean_from_proba(proba_list):
     """