Update README.md

README.md

@@ -9,7 +9,7 @@ metrics:
 base_model:
 - microsoft/DialoGPT-small
 pipeline_tag: text-generation
-library_name: diffusers
+library_name: transformers
 tags:
 - text-generation-inference
 ---
@@ -51,3 +51,16 @@ Based on initial testing, the model may exhibit repetitive text generation, espe
 ### Recommendations
 
 Users should be aware of the model's limitations in generating coherent long text and potential biases. It is recommended to experiment with different generation parameters (`max_length`, `no_repeat_ngram_size`, sampling strategies) to improve output quality. For any critical applications, thorough testing and human review of generated content are essential.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model using the transformers library.
+```python
+from transformers import pipeline
+generator = pipeline("text-generation", model="anktechsol/ankiGPT-small")
+
+prompt = "Write a short story about a day in the life of a student in a bustling Indian city, describing their commute, interactions at school, and a cultural event they attend in the evening."
+generated_text = generator(prompt, max_length=300, num_return_sequences=1)
+
+print(generated_text[0]['generated_text'])
+```