sorizano commited on
Commit
9b95e7e
verified
1 Parent(s): abc9ce9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -1,17 +1,23 @@
1
  import streamlit as st
2
- from transformers import pipline
3
 
 
4
  st.title("Generaci贸n de Texto con Hugging Face")
5
  st.write("Introduce un texto para que el modelo lo complete:")
6
 
 
7
  user_input = st.text_area("Texto de entrada")
8
 
9
- text_generator = pipline("text-generation")
 
10
 
 
11
  if user_input:
12
  generated_text = text_generator(user_input, max_length=50, num_return_sequences=1)
13
-
14
- st.write("Texto Generado:")
 
15
  st.write(generated_text[0]['generated_text'])
16
 
17
- st.info("Este es un ejmemplo b谩sico de generaci贸n de texto utilizando un modelo preentrenado de Hugging Face.")
 
 
1
  import streamlit as st
2
+ from transformers import pipeline
3
 
4
+ # Configuraci贸n de la aplicaci贸n de Streamlit
5
  st.title("Generaci贸n de Texto con Hugging Face")
6
  st.write("Introduce un texto para que el modelo lo complete:")
7
 
8
+ # Entrada de texto del usuario
9
  user_input = st.text_area("Texto de entrada")
10
 
11
+ # Cargar el modelo de generaci贸n de texto
12
+ text_generator = pipeline("text-generation", model="flax-community/gpt-2-spanish")
13
 
14
+ # Generar texto basado en la entrada del usuario
15
  if user_input:
16
  generated_text = text_generator(user_input, max_length=50, num_return_sequences=1)
17
+
18
+ # Mostrar el resultado de la generaci贸n de texto
19
+ st.write("Texto generado:")
20
  st.write(generated_text[0]['generated_text'])
21
 
22
+ # Mensaje de informaci贸n
23
+ st.info("Este es un ejemplo b谩sico de generaci贸n de texto utilizando un modelo preentrenado de Hugging Face.")