Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,25 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
ejemplos = [["El cielo es de color azul"],["Los estudiantes de sistemas"],["Los elefantes son rosas"]]
|
4 |
|
5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
+
from transformers import pipeline
|
3 |
|
4 |
+
# Cargar el modelo de generaci贸n de texto en espa帽ol
|
5 |
+
generador = pipeline("text-generation", model="datificate/gpt2-small-spanish")
|
6 |
+
|
7 |
+
# Funci贸n de generaci贸n de texto
|
8 |
+
def generar_texto(input_text):
|
9 |
+
resultado = generador(input_text, max_length=50, num_return_sequences=1)
|
10 |
+
return resultado[0]['generated_text']
|
11 |
+
|
12 |
+
# Ejemplos de entrada
|
13 |
ejemplos = [["El cielo es de color azul"],["Los estudiantes de sistemas"],["Los elefantes son rosas"]]
|
14 |
|
15 |
+
# Interfaz de Gradio
|
16 |
+
interfaz = gr.Interface(
|
17 |
+
fn=generar_texto,
|
18 |
+
inputs="text",
|
19 |
+
outputs="text",
|
20 |
+
examples=ejemplos,
|
21 |
+
title="Generador de Textos en Espa帽ol",
|
22 |
+
description="Introduce una frase o selecciona un ejemplo para que el modelo complete el texto."
|
23 |
+
)
|
24 |
+
|
25 |
+
interfaz.launch()
|