Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -169,25 +169,34 @@ with zero_shot:
     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"
+        st.caption("Usando Clip de OpenAI y Convnext de Facebook")
 
         labels_for_classification =  ["A yellow deep fried smashed plantain", 
                             "Fried food", 
                             "Anything"]
-    
+
+
+        model_dict = {
+            'Clip (openai/clip-vit-large-patch14-336)' : 'frncscp/patacoptimus-prime',
+            'Convnext (facebook/convnext-large-224)' : 'frncscp/pataconxt'
+        }
+        
+        model_choice = st.multiselect("Seleccione uno o varios modelos de clasificación", model_dict.keys(), key = 'zs_multiselect')
+
+        
         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 33.33%)', 0, 100, 34, key = 'threshold_ZS')
-
+    
     with col_b:
 
         
         if st.button(key = 'ZS_button', label ='¿Hay un patacón en la imagen?'):
-            if uploaded_file is not None:
+            if len(model_choice) != 1:
+                print('Recuerda seleccionar un solo modelo de clasificación')
+            elif uploaded_file is not None:
                 if not zsloaded:
                     with st.spinner("Cargando modelo de clasificación..."):
-                        classifier = pipeline("zero-shot-image-classification", model = clip)
+                        classifier = pipeline("zero-shot-image-classification", model = model_choice[0])
                         zsloaded = True
                 with st.spinner('Cargando predicción...'):
                     img = preprocess(uploaded_file, module = 'pil')