import gradio as gr
from transformers import pipeline

generator = pipeline("summarization", model="kriton/greek-text-summarization")


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

iface = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Remote LLM Text Generation",
    description="Enter a prompt to generate text from the model."
)

# Launch the Gradio interface
iface.launch()