Update README.md

README.md

@@ -33,11 +33,11 @@ Use the code below to get started with the model.
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 # Load the tokenizer and model
-tokenizer = T5Tokenizer.from_pretrained("john_doe/fine-tuned-t5-small")
-model = T5ForConditionalGeneration.from_pretrained("john_doe/fine-tuned-t5-small")
+tokenizer = T5Tokenizer.from_pretrained("imranali291/flux-prompt-enhancer")
+model = T5ForConditionalGeneration.from_pretrained("imranali291/flux-prompt-enhancer")
 
 # Example input
-input_text = "Translate English to French: The cat is on the table."
+input_text = "Futuristic cityscape at twilight descent."
 
 # Tokenize input
 input_ids = tokenizer(input_text, return_tensors="pt").input_ids