AlejandraFlorezM commited on
Commit
4d820fd
·
verified ·
1 Parent(s): 3ec3c44

Delete app.py

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