Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -7,6 +7,8 @@ import os
 import cv2
 import numpy as np
 
+zsloaded = False
+
 st.set_page_config(
     page_title = 'Patacotrón',
     layout = 'wide',
@@ -132,16 +134,13 @@ with vit:
         uploaded_file = st.file_uploader(key = 'ViT_upload', label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
         flag = False
         threshold = st.slider('¿Cuál va a ser el límite desde donde se considere patacón? (se recomienda por encima del 80%)', 0, 100, 80, key = 'threshold_vit')
-        if st.button(key = 'ViT_button', label ='¿Hay un patacón en la imagen?'):
-            if uploaded_file is not None:
-                with st.spinner('Cargando predicción...'):
-                    classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
-                    img = preprocess(uploaded_file, module = 'pil')
-                    flag = True
-            else:
-                st.write("Asegúrate de haber subido correctamente la imagen.")
+
     with col_b:
-        if flag:
+
+    if st.button(key = 'ViT_button', label ='¿Hay un patacón en la imagen?') and uploaded_file is not None:
+        with st.spinner('Cargando predicción...'):
+            classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
+            img = preprocess(uploaded_file, module = 'pil')
             classifier = classifier(img)          
             y_gorrito = classifier[0]["score"]
             if round(float(y_gorrito*100)) >= threshold:
@@ -150,10 +149,14 @@ with vit:
                 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("Asegúrate de haber subido correctamente la imagen.")
+
+
                 
 
 with zero_shot:
-
+    
     col_a, col_b = st.columns(2)
 
     with col_a:
@@ -172,24 +175,16 @@ with zero_shot:
 
     with col_b:
 
-        loaded = False
         
-        if st.button(key = 'ZS_button', label ='¿Hay un patacón en la imagen?'):
-            if not loaded:
+        if st.button(key = 'ZS_button', label ='¿Hay un patacón en la imagen?') and uploaded_file is not None:
+            if not zsloaded:
                 with st.spinner("Cargando modelo de clasificación..."):
                     classifier = pipeline("zero-shot-image-classification", model = clip)
-                    loaded = True
-            if uploaded_file is not None:
-                with st.spinner('Cargando predicción...'):
-                    img = preprocess(uploaded_file, module = 'pil')
-                    flag = True
-            else:
-                st.write("Asegúrate de haber subido correctamente la imagen.")
-        
-        if flag:
-
-            zs_classifier = classifier(img, 
-                    candidate_labels = labels_for_classification)
+                    zsloaded = True
+            with st.spinner('Cargando predicción...'):
+                img = preprocess(uploaded_file, module = 'pil')
+                        zs_classifier = classifier(img, 
+                candidate_labels = labels_for_classification)
 
             for clase in zs_classifier:
                 if clase['label'] == 'A yellow deep fried smashed plantain':
@@ -201,6 +196,10 @@ with zero_shot:
                 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("Asegúrate de haber subido correctamente la imagen.")
+        
+
         
 with autoencoder:
     st.write('Próximamente')