triptuner / README.md
yoonusajward01's picture
Update README.md
00f257f verified
metadata
library_name: transformers
tags:
  - gpt2
  - transformers
  - text-generation
  - adventure
  - travel-itinerary
  - custom-model
pipeline_tag: text-generation

TripTuner: Adventure Travel Itinerary Generator for Central Province, Sri Lanka

Overview

TripTuner is a custom-trained GPT-2 model designed to generate personalized adventure travel itineraries specifically for locations within the Central Province of Sri Lanka. The model is fine-tuned to provide detailed descriptions of adventure activities such as trekking, hiking, camping, and exploring scenic views, making it ideal for travel enthusiasts and tour planners.

Model Details

  • Model Type: GPT-2
  • 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

You can use this model directly with the Hugging Face Inference API or load it into your Python environment using the Transformers library.

Using the Inference API

You can test the model directly via the Hugging Face platform by clicking on the "Inference API" tab.

Using Transformers Pipeline

from transformers import pipeline

# Load the text generation pipeline using the uploaded model from Hugging Face
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, truncation=True)
print(response[0]['generated_text'])