app.py

import gradio as gr
from transformers import pipeline

# Load your Hugging Face model (replace with your exact model path)
generator = pipeline("text-generation", model="vkaigen/Gpoetry")

def generate_poem(prompt):
    results = generator(prompt, max_length=100, num_return_sequences=1)
    return results[0]["generated_text"]

gr.Interface(
    fn=generate_poem,
    inputs=gr.Textbox(lines=4, placeholder="Write the beginning of a Georgian poem..."),
    outputs="text",
    title="📝 Georgian Poetry Generator",
    description="Type a prompt to generate beautiful Georgian poetry using your fine-tuned GPT-2 model."
).launch()