AlejandraFlorezM commited on
Commit
86530e0
·
verified ·
1 Parent(s): 4d820fd

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ # Título de la app
5
+ st.title("Generador de Texto Quinta Optimization")
6
+
7
+ # Cargar modelo en español optimizado
8
+ @st.cache_resource
9
+ def load_model():
10
+ return pipeline("text-generation", model="datificate/gpt2-small-spanish")
11
+
12
+ generator = load_model()
13
+
14
+ # Entrada del usuario
15
+ st.subheader("Entrada")
16
+ prompt = st.text_area("Escribe un texto para que el modelo continúe (en español):")
17
+
18
+ # Generar respuesta
19
+ if st.button("Generar Texto"):
20
+ if not prompt.strip():
21
+ st.warning("Por favor, ingresa un texto válido.")
22
+ else:
23
+ with st.spinner("Generando texto..."):
24
+ result = generator(
25
+ prompt,
26
+ max_length=30, # Generación más corta
27
+ temperature=0.2, # Baja creatividad
28
+ top_p=0.8, # Sampling basado en probabilidad acumulativa
29
+ top_k=50, # Limitar a las 50 palabras más probables
30
+ repetition_penalty=1.3, # Penalizar repeticiones
31
+ num_return_sequences=1,
32
+ eos_token_id=50256 # Detener en fin de frase
33
+ )
34
+ response = result[0]["generated_text"]
35
+ st.subheader("Texto Generado:")
36
+ st.write(response)