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-it-en")
-model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-it-en")
+tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-it-en")
+model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-it-en")
 
 def generate_response(input_text):
     input_ids = tokenizer("translate Italian to English:" + input_text, return_tensors="pt").input_ids
@@ -49,8 +49,8 @@ import spacy
 
 nlp = spacy.load("it_core_news_sm")
 
-tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-it-en")
-model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/mt0-Quadrifoglio-mt-it-en")
+tokenizer = AutoTokenizer.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-it-en")
+model = AutoModelForSeq2SeqLM.from_pretrained("LeonardPuettmann/Quadrifoglio-mt-it-en")
 
 def generate_response(input_text):
     input_ids = tokenizer("translate Italian to English: " + input_text, return_tensors="pt").input_ids