manvik28 commited on
Commit
00db5d4
·
1 Parent(s): e161321

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def chunk_text_to_window_size_and_predict_proba(input_ids, attention_mask, total
56
 
57
  start = end
58
 
59
-
60
 
61
  def get_sentiment(tokens):
62
  outputs = model(**tokens)
 
56
 
57
  start = end
58
 
59
+ return proba_list
60
 
61
  def get_sentiment(tokens):
62
  outputs = model(**tokens)