Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -157,11 +157,47 @@ with vit:
                 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.caption(f'La probabilidad de que la imagen tenga un patacón es del: {round(float(y_gorrito * 100), 2)}%')  
                 
 
 with zero_shot:
-    st.write('Próximamente')
+
+    col_a, col_b = st.columns(2)
+
+    with col_a:
+    
+        st.title("Clasificación Zero-Shot")
+        st.caption("Usando [openai/clip-vit-large-patch14-336](https://huggingface.co/openai/clip-vit-large-patch14-336)")
+        
+        clip = "openai/clip-vit-large-patch14-336"
+        classifier = pipeline("zero-shot-image-classification", model = clip)
+
+        labels_for_classification =  ["A yellow deep fried smashed plantain", 
+                            "Fried food", 
+                            "Anything"]
+    
+        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 80%)', 0, 100, 80, key = 'threshold_ZS')
+
+        if st.button(key = 'ZS_button', label ='¿Hay un patacón en la imagen?'):
+        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.")
+
+
+    with col_b:
+        if flag:
+            y_gorrito = (classifier(img, 
+                                candidate_labels = labels_for_classification)[1]["score"])
+            if round(float(y_gorrito*100)) >= threshold:
+                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)}%')  
+