Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ def sentiment_analysis(sentence, model_name):
|
|
| 40 |
model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-zh-en")
|
| 41 |
|
| 42 |
translated_tokens = model.generate(
|
| 43 |
-
**tokenizer(
|
| 44 |
)
|
| 45 |
# for t in translated:
|
| 46 |
# print( tokenizer.decode(t, skip_special_tokens=True) )
|
|
|
|
| 40 |
model = AutoModelForSeq2SeqLM.from_pretrained("Helsinki-NLP/opus-mt-zh-en")
|
| 41 |
|
| 42 |
translated_tokens = model.generate(
|
| 43 |
+
**tokenizer(sentences, return_tensors="pt", padding=True)
|
| 44 |
)
|
| 45 |
# for t in translated:
|
| 46 |
# print( tokenizer.decode(t, skip_special_tokens=True) )
|