JohnKouf's picture
Create app.py
945bbd0 verified
raw
history blame
514 Bytes
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()