AlejandraFlorezM commited on
Commit
3ec3c44
·
verified ·
1 Parent(s): 52dedc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
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 Optimización")
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
- result = generator(
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