Spaces:

Himanshusingh
/

sec_sentiment_analysis

Running

Himanshusingh commited on Jul 23, 2023

Commit

27e4a2e

1 Parent(s): 37fc349

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,6 +65,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)
 def get_mean_from_proba(proba_list):
     """

         probabilities = torch.nn.functional.softmax(outputs[0], dim = -1)
         proba_list.append(probabilities)
+        return proba_list
 def get_mean_from_proba(proba_list):
     """