Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def get_sentiment_model():
|
|
24 |
def get_sentiment(text):
|
25 |
input_ids = sentiment_tokenizer .encode(text + '</s>', return_tensors='pt')
|
26 |
output = sentiment_model.generate(input_ids=input_ids,max_length=2)
|
27 |
-
dec = [
|
28 |
label = dec[0]
|
29 |
return label
|
30 |
|
|
|
24 |
def get_sentiment(text):
|
25 |
input_ids = sentiment_tokenizer .encode(text + '</s>', return_tensors='pt')
|
26 |
output = sentiment_model.generate(input_ids=input_ids,max_length=2)
|
27 |
+
dec = [sentiment_tokenizer.decode(ids) for ids in output]
|
28 |
label = dec[0]
|
29 |
return label
|
30 |
|