Spaces:

frncscp
/

Patacotron

Running

frncscp commited on May 20, 2023

Commit

3a93dd5

1 Parent(s): 4febb7b

Update pages/Entorno de Ejecución.py

Files changed (1) hide show

pages/Entorno de Ejecución.py CHANGED Viewed

@@ -230,7 +230,7 @@ with zero_shot:
                     zs_classifier = classifier(img,
                     candidate_labels = labels_for_classification)
-                    label, _, y_gorrito = multiclass_prediction(zs_classifier)
                     if label == "A yellow deep fried smashed plantain":
                         st.success("¡Patacón Detectado!")

                     zs_classifier = classifier(img,
                     candidate_labels = labels_for_classification)
+                    label, _, y_gorrito = multiclass_prediction(zs_classifier, labels_for_classification[0])
                     if label == "A yellow deep fried smashed plantain":
                         st.success("¡Patacón Detectado!")