puettmann
/

Quadrifoglio-mt-it-en

text2text-generation

Model card Files Files and versions

Leonard Püttmann commited on Dec 30, 2024

Commit

d45e460

·

verified ·

1 Parent(s): 309e83a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-
 model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-it-en")
 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)

 model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-it-en")
 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)