Spaces:
Runtime error
Runtime error
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() |