Update README.md

README.md

@@ -27,8 +27,8 @@ Quadrifoglio is an encoder-decoder transformer model for English-Italian text tr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 # Load model and tokenizer from checkpoint directory
-tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-en-it")
-model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-en-it")
+tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-en-it")
+model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-en-it")
 
 def generate_response(input_text):
     input_ids = tokenizer("translate English to Italian:" + input_text, return_tensors="pt").input_ids
@@ -50,8 +50,8 @@ import spacy
 
 nlp = spacy.load("en_core_web_sm")
 
-tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-en-it")
-model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-en-it")
+tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-en-it")
+model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-en-it")
 
 def generate_response(input_text):
     input_ids = tokenizer("translate Italian to English: " + input_text, return_tensors="pt").input_ids