puettmann
/

Foglietta-mt-en-it

text2text-generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Leonard Püttmann commited on 8 days ago

Commit

5a2bf9e

·

verified ·

1 Parent(s): 8219f8b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/Foglietta-mt-en-it")
 model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/Foglietta-mt-en-it")
 def generate_response(input_text):
-    input_ids = tokenizer("translate Italian to English: " + input_text, return_tensors="pt").input_ids
     output = model.generate(input_ids, max_new_tokens=256)
     return tokenizer.decode(output[0], skip_special_tokens=True)

 model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/Foglietta-mt-en-it")
 def generate_response(input_text):
+    input_ids = tokenizer("translate English to Italian: " + input_text, return_tensors="pt").input_ids
     output = model.generate(input_ids, max_new_tokens=256)
     return tokenizer.decode(output[0], skip_special_tokens=True)