ukmastercom/itinera-llm-agent

1import os2import gradio as gr3from transformers import pipeline4 5# ✅ Set Hugging Face cache directory (prevents permission issues)6os.environ["HF_HOME"] = "/tmp/models"7os.environ["TRANSFORMERS_CACHE"] = "/tmp/models"8os.environ["HF_DATASETS_CACHE"] = "/tmp/models"9os.environ["HF_METRICS_CACHE"] = "/tmp/models"10 11# ✅ Load model (will use cached model if already downloaded)12generator = pipeline(13    "text-generation",14    model="distilgpt2",15    cache_dir="/tmp/models"16)17 18# ✅ Function to generate text from prompt19def generate_text(prompt):20    output = generator(prompt, max_length=200, num_return_sequences=1)21    return output[0]["generated_text"]22 23# ✅ Gradio interface24iface = gr.Interface(25    fn=generate_text,26    inputs=gr.Textbox(lines=3, placeholder="Enter your prompt here..."),27    outputs="text",28    title="GPT Trip Planner",29    description="Generate travel itineraries using GPT."30)31 32# ✅ Launch app (Spaces will auto-run this)33if __name__ == "__main__":34    iface.launch(server_name="0.0.0.0", server_port=7860)35