Spaces:

ieuniversity
/

Sciencebrief_translation

Runtime error

chayyyy commited on Apr 12, 2023

Commit

9f2d4e6

•

1 Parent(s): 0f3cc9c

Update app.py (#1)

- Update app.py (7c6189a2e98a0e1c45f5d473c6e81c32da027b4f)

Co-authored-by: mes <chayyyy@users.noreply.huggingface.co>

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ tokenizer = AutoTokenizer.from_pretrained("ieuniversity/sciencebrief_translation
 model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_translation")
 def translate(text):
-    input_ids = tokenizer.encode(text, return_tensors="pt")
     output_ids = model.generate(input_ids)
     output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
     return output_text

 model = AutoModelForSeq2SeqLM.from_pretrained("ieuniversity/sciencebrief_translation")
 def translate(text):
+    input_ids = tokenizer(text, return_tensors="pt", truncation=True)['input_ids']
     output_ids = model.generate(input_ids)
     output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
     return output_text