Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -141,8 +141,14 @@ with vit:
             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"]
+                
+                classifier = classifier(img) 
+
+                for clase in classifier:
+                    if clase['label'] == 'Patacón-True':
+                        y_gorrito = clase["score"]
+                
+                #y_gorrito = classifier[0]["score"]
                 if round(float(y_gorrito*100)) >= threshold:
                     st.success("¡Patacón Detectado!")
                 else:
@@ -188,7 +194,7 @@ with zero_shot:
 
                 for clase in zs_classifier:
                     if clase['label'] == 'A yellow deep fried smashed plantain':
-                        y_gorrito = (clase["score"])
+                        y_gorrito = clase["score"]
                         
                 if round(float(y_gorrito*100)) >= threshold:
                     st.success("¡Patacón Detectado!")