Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ import time
|
|
4 |
|
5 |
#Nombre_modelo = 'pysentimiento/robertuito-sentiment-analysis'
|
6 |
#######################creacion de funciones######################
|
|
|
7 |
def cargar_modelo(model_name):
|
8 |
return pipeline("zero-shot-classification", model=model_name, device=0)
|
9 |
|
@@ -30,7 +31,7 @@ model_options = [
|
|
30 |
# Seleccionar el modelo del pipeline
|
31 |
selected_model = st.selectbox("Selecciona un modelo", model_options)
|
32 |
#funcion eleccion del modelo
|
33 |
-
|
34 |
|
35 |
#asignacion del modelo a usar
|
36 |
classifier = cargar_modelo(selected_model)
|
|
|
4 |
|
5 |
#Nombre_modelo = 'pysentimiento/robertuito-sentiment-analysis'
|
6 |
#######################creacion de funciones######################
|
7 |
+
@st.cache_resource
|
8 |
def cargar_modelo(model_name):
|
9 |
return pipeline("zero-shot-classification", model=model_name, device=0)
|
10 |
|
|
|
31 |
# Seleccionar el modelo del pipeline
|
32 |
selected_model = st.selectbox("Selecciona un modelo", model_options)
|
33 |
#funcion eleccion del modelo
|
34 |
+
|
35 |
|
36 |
#asignacion del modelo a usar
|
37 |
classifier = cargar_modelo(selected_model)
|