Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -194,12 +194,11 @@ with zero_shot:
             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.')
         
         if st.button(key = 'ZS_button', label ='¿Hay un patacón en la imagen?'):
-            if uploaded_file is not None:
-                if not zsloaded:  
-                        st.write('Primero se necesita cargar el modelo.')
-                else:
+            if uploaded_file is not None:     
+                if zsloaded:
                     with st.spinner('Cargando predicción...'):
                         img = preprocess(uploaded_file, module = 'pil')
                         zs_classifier = classifier(img, 
@@ -213,8 +212,11 @@ with zero_shot:
                             st.success("¡Patacón Detectado!")
                         else:
                             st.error("No se considera que haya un patacón en la imagen")
+                            
                         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.')
             else:
                 st.write("Asegúrate de haber subido correctamente la imagen.")