Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -138,6 +138,7 @@ with vit:
         st.title('Visual Transformers')
         st.caption('One class is all you need!')
         uploaded_file = st.file_uploader(key = 'ViT_upload', label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
+        flag = False
         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...'):
@@ -145,11 +146,13 @@ with vit:
                     img = preprocess(uploaded_file, module = 'pil')
                     raw_img = img
                     img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))     
+                    flag = True
             else:
                 st.write("Asegúrate de haber subido correctamente la imagen.")
     with col_b:
-        st.write(classifier(img))
-        st.image(raw_img)
+        if flag:
+            st.write(classifier(img))
+            st.image(raw_img)
 
 with zero_shot:
     st.write('Próximamente')