PanquequeG commited on
Commit
8d7f26f
verified
1 Parent(s): 27278a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -21
app.py CHANGED
@@ -1,25 +1,5 @@
1
  import gradio as gr
2
- 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()
 
1
  import gradio as gr
 
2
 
 
 
 
 
 
 
 
 
 
3
  ejemplos = [["El cielo es de color azul"],["Los estudiantes de sistemas"],["Los elefantes son rosas"]]
4
 
5
+ gr.load("huggingface/datificate/gpt2-small-spanish", examples=ejemplos).launch()