Update README.md

README.md

@@ -24,8 +24,8 @@ To use it with transformers library take a look into the following code snippet:
 # pip install accelerate
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
-tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
-model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large", device_map="auto")
+tokenizer = T5Tokenizer.from_pretrained("knowledgator/flan-t5-small-for-classification")
+model = T5ForConditionalGeneration.from_pretrained("knowledgator/flan-t5-small-for-classification", device_map="auto")
 
 input_text = "Define sentiment of the following text: I love to travel and someday I will see the world."
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids.to("cuda")