Update README.md

README.md

@@ -64,4 +64,24 @@ The following hyperparameters were used during training:
 - Transformers 4.43.1
 - Pytorch 2.1.2+cu121
 - Datasets 2.19.1
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1
+
+## Test Model Code
+
+```python
+model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
+
+enhancer = pipeline('text2text-generation',
+                    model=model,
+                    tokenizer=tokenizer,
+                    repetition_penalty= 1.2,
+                    device=device)
+
+max_target_length = 256
+prefix = "enhance prompt: "
+
+short_prompt = "beautiful house with text 'hello'"
+answer = enhancer(prefix + short_prompt, max_length=max_target_length)
+final_answer = answer[0]['generated_text']
+print(final_answer)
+```