cccc commited on
Commit
ca1a3ec
·
1 Parent(s): 5d5c2d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(src_text, return_tensors="pt", padding=True)
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) )