Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|