Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -141,12 +141,12 @@ with vit:
         uploaded_file = st.file_uploader(key = 'ViT_upload', label = 'Sube la imagen a clasificar',type= ['jpg','png', 'jpeg', 'jfif', 'webp', 'heic'])
         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...', key = 'vitspinner')
-                classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
-                img = preprocess(uploaded_file, module = 'pil')
-                raw_img = img
-                img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))     
-                classifier(img)
+                with st.spinner('Cargando predicción...', key = 'vitspinner'):
+                    classifier = pipeline("image-classification", model="frncscp/patacoptimus-prime")
+                    img = preprocess(uploaded_file, module = 'pil')
+                    raw_img = img
+                    img = cv2.resize(img, (IMAGE_WIDTH, IMAGE_HEIGHT))     
+                    classifier(img)
             else:
                 st.write("Asegúrate de haber subido correctamente la imagen.")