Update README.md

README.md

@@ -12,6 +12,15 @@ pipeline_tag: text2text-generation
 ```python
 from transformers import pipeline
 
+# Model checkpoint
+model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
+
+# Tokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
+
+# Model
+model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
+
 enhancer = pipeline('text2text-generation', model=model, tokenizer=tokenizer, repetition_penalty=1.5, device="cuda")
 max_target_length = 256
 prefix = "enhance prompt: "