Spaces:

frncscp
/

Patacotron

Sleeping

frncscp commited on Nov 26, 2023

Commit

baa95fe

1 Parent(s): f6ec741

Update pages/Entorno de Ejecución.py

Files changed (1) hide show

pages/Entorno de Ejecución.py CHANGED Viewed

@@ -325,7 +325,7 @@ with classic_ml:
                 with st.spinner('Cargando predicción...'):
                     img = preprocess(uploaded_file)
-                    selected_models = [joblib.load(model_dict[model_name]) for model_name in model_choice if model_name not in selected_models]
                     y_gorrito, raw_img =  _pca_predict(selected_models, img)
                     if round(float(y_gorrito*100)) >= threshold:

                 with st.spinner('Cargando predicción...'):
                     img = preprocess(uploaded_file)
+                    selected_models = [joblib.load(model_dict[model_name]) for model_name in model_choice]
                     y_gorrito, raw_img =  _pca_predict(selected_models, img)
                     if round(float(y_gorrito*100)) >= threshold: