Update app.py
Browse files
app.py
CHANGED
@@ -2,13 +2,16 @@ import streamlit as st
|
|
2 |
import torch
|
3 |
from transformers import pipeline
|
4 |
|
|
|
|
|
|
|
5 |
# Título de la app
|
6 |
-
st.title("Generador de Texto Quinta
|
7 |
|
8 |
# Cargar modelo en español optimizado
|
9 |
@st.cache_resource
|
10 |
def load_model():
|
11 |
-
return pipeline("text-generation", model="datificate/gpt2-small-spanish")
|
12 |
|
13 |
generator = load_model()
|
14 |
|
@@ -22,7 +25,7 @@ if st.button("Generar Texto"):
|
|
22 |
st.warning("Por favor, ingresa un texto válido.")
|
23 |
else:
|
24 |
with st.spinner("Generando texto..."):
|
25 |
-
|
26 |
prompt,
|
27 |
max_length=30, # Generación más corta
|
28 |
temperature=0.2, # Baja creatividad
|
|
|
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 |
|
|
|
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
|