Spaces:

frncscp
/

Patacotron

Running

frncscp commited on Nov 26, 2023

Commit

da6bf3e

1 Parent(s): e4ba9ea

Update pages/Entorno de Ejecución.py

Files changed (1) hide show

pages/Entorno de Ejecución.py CHANGED Viewed

@@ -47,7 +47,8 @@ def _pca_predict(models, _img):
     fl_img =[img.flatten()]
     data = pca.transform(fl_img)
     for model in models:
-        y_gorrito += model.predict_proba(data)[0][Categories.index("Patacon-True")]
     return [y_gorrito / len(models), raw_img]
 #def classic_ml_prediction(clfs, _img):

     fl_img =[img.flatten()]
     data = pca.transform(fl_img)
     for model in models:
+        prediction = model.predict_proba(data)
+        y_gorrito += prediction[0][Categories.index("Patacon-True")]
     return [y_gorrito / len(models), raw_img]
 #def classic_ml_prediction(clfs, _img):