AlejandraFlorezM commited on
Commit
fd31c78
·
verified ·
1 Parent(s): 68ce504

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -37
app.py DELETED
@@ -1,37 +0,0 @@
1
- import streamlit as st
2
- from transformers import pipeline
3
-
4
- # Título de la app
5
- st.title("Generador de Texto Séptima 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=20, # Generación más corta
27
- temperature=0.2, # Baja creatividad
28
- top_p=0.7, # Sampling basado en probabilidad acumulativa
29
- top_k=20, # Limitar a las 50 palabras más probables
30
- repetition_penalty=1.5, # Penalizar repeticiones
31
- num_return_sequences=1,
32
- eos_token_id=50256 # Detener en fin de frase
33
- do_sample=False # Genración determinista
34
- )
35
- response = result[0]["generated_text"]
36
- st.subheader("Texto Generado:")
37
- st.write(response)