Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -22,7 +22,8 @@ st.sidebar.write("contact@patacotron.tech")
 
 cnn, vit, zero_shot, autoencoder, svm, iforest, gan = st.tabs(["CNN", "ViT", "Zero-Shot", "Autoencoder", "OC-SVM", 'iForest', 'GAN'])
 
-def predict(model_list, weights, img): #for non-supported formats
+@st.cache_resource
+def predict(model_list, weights, img): 
     y_gorrito = 0
     raw_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))
@@ -38,6 +39,13 @@ def preprocess(file_uploader, module = 'cv2'): #makes the uploaded image readabl
         img = Image.open(file_uploader)
     return img
 
+@st.cache_resource
+def load_clip():
+    classifier = pipeline("zero-shot-image-classification", model = 'openai/clip-vit-large-patch14-336')
+    zsloaded = True
+    st.write('El modelo ha sido cargado, no es necesario volver a presionar el botón.')
+    return classifier
+
 with cnn:
 
     col_a, col_b, = st.columns(2)
@@ -187,15 +195,14 @@ with zero_shot:
         
         uploaded_file = st.file_uploader(key = 'ZS_upload', label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
         threshold = st.slider('¿Cuál va a ser el límite desde donde se considere patacón? (se recomienda por encima del 33.33%)', 0, 100, 34, key = 'threshold_ZS')
-    
+        
     with col_b:
 
         if st.button(key = 'clip_button', label = 'Cargar modelo de clasificación (puede tomar algo de tiempo)'):
             with st.spinner("Cargando modelo de clasificación..."):
-                    classifier = pipeline("zero-shot-image-classification", model = 'openai/clip-vit-large-patch14-336')
-                    zsloaded = True
-                    st.write('El modelo ha sido cargado, no es necesario volver a presionar el botón.')
-        
+                classifier = load_clip()
+                zsloaded = True
+                
         if st.button(key = 'ZS_button', label ='¿Hay un patacón en la imagen?'):
             if uploaded_file is not None:     
                 if zsloaded:
@@ -216,7 +223,7 @@ with zero_shot:
                         st.caption(f'La probabilidad de que la imagen tenga un patacón es del: {round(float(y_gorrito * 100), 2)}%')  
                         st.image(img)
                 else:
-                    st.write('Primero se necesita cargar el modelo.')
+                    st.write('Primero se necesita cargar el modelo.', zsloaded)
             else:
                 st.write("Asegúrate de haber subido correctamente la imagen.")