import gradio as gr
from transformers import pipeline

generator = pipeline('text-generation', model='Emilianohack6950/M.A.R.I.A')

def generate_text(prompt, tokens_to_generate):
    text = generator(prompt, max_length=tokens_to_generate)
    return text[0]['generated_text']

gr.Interface(
    generate_text,
    inputs=["text", "number"],
    outputs="text",
    title="Generador de texto GPT-2",
    description="Genera texto basado en un prompt dado utilizando el modelo pre-entrenado de GPT-2.",
).launch()