Update app.py

app.py

@@ -6,8 +6,10 @@ from transformers import AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("Oysiyl/elvish-translator-quenya-t5-small")
 model = AutoModelForSeq2SeqLM.from_pretrained("Oysiyl/elvish-translator-quenya-t5-small")
 
+prefix = "translate English to Elvish: "
+
 def greet(name):
-    inputs = tokenizer(text, return_tensors="pt").input_ids
+    inputs = tokenizer(prefix + text, return_tensors="pt").input_ids
     outputs = model.generate(inputs, max_new_tokens=40, do_sample=True, top_k=30, top_p=0.95)
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return result