Views
No views yet
1from transformers import pipeline
2
3# Load the text generation pipeline using the uploaded model from Hugging Face
4generator = pipeline('text-generation', model='yoonusajward01/triptuner')
5
6# Test the model with a prompt
7response = generator("[Q] Describe an adventure itinerary for Knuckles Mountain Range.", max_length=150, num_return_sequences=1)
8print(response[0]['generated_text'])