Update README.md

README.md

@@ -22,6 +22,7 @@ pipeline_tag: text-generation
 - **Library**: Transformers by Hugging Face
 - **Use Case**: Text generation focused on adventure travel itineraries
 - **Languages**: English
+- **Base Model**: GPT-2
 - **Training Data**: The model was fine-tuned on a custom dataset of adventure activities in various locations within Sri Lanka's Central Province.
 
 ## How to Use
@@ -41,5 +42,5 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='yoonusajward01/triptuner')
 
 # Test the model with a prompt
-response = generator("[Q] Describe an adventure itinerary for Knuckles Mountain Range.", max_length=150, num_return_sequences=1)
+response = generator("[Q] Describe an adventure itinerary for Knuckles Mountain Range.", max_length=150, num_return_sequences=1, truncation=True)
 print(response[0]['generated_text'])