Spaces:

somosnlp-hackathon-2022
/

Spanish-Nahuatl-Translation

Running

rockdrigoma commited on Mar 31, 2022

Commit

b6287db

•

1 Parent(s): 956e550

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained('hackathon-pln-es/t5-small-spanish
 tokenizer = AutoTokenizer.from_pretrained('hackathon-pln-es/t5-small-spanish-nahuatl')
 def predict(input):
-  input_ids = tokenizer('translate Spanish to Nahuatl: ' + sentence, return_tensors='pt').input_ids
   outputs = model.generate(input_ids)
   outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
   return outputs

 tokenizer = AutoTokenizer.from_pretrained('hackathon-pln-es/t5-small-spanish-nahuatl')
 def predict(input):
+  input_ids = tokenizer('translate Spanish to Nahuatl: ' + input, return_tensors='pt').input_ids
   outputs = model.generate(input_ids)
   outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0]
   return outputs