Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -97,22 +97,20 @@ with cnn:
                         if not executed: 
                             ultraptctrn = [load_model(model_dict[model]) for model in ultraversions]
                             executed = True
-                            final_weights = weight_list if len(weight_list) >= 1 else [1 for i in range(len(ultraptctrn))]
+                        final_weights = weight_list if len(weight_list) >= 1 else [1 for i in range(len(ultraptctrn))]
                         try:
-                            y_gorrito, raw_img = tf_predict(ultraptctrn, weight_list, img)
+                            y_gorrito, raw_img = tf_predict(ultraptctrn, final_weights, img)
                         except:
-                            y_gorrito, raw_img = basic_predict(ultraptctrn, weight_list, img)
+                            y_gorrito, raw_img = basic_predict(ultraptctrn, final_weights, img)
                         
                 else:
                     with st.spinner('Cargando predicción...'):
                         selected_models = [load_model(model_dict[model]) for model in model_choice if model not in selected_models]
                         final_weights = weight_list if len(weight_list) >= 1 else [1 for i in range(len(selected_models))]
                         try:             
-                            y_gorrito, raw_img = tf_predict(selected_models, img)
+                            y_gorrito, raw_img = tf_predict(selected_models, final_weights, img)
                         except:
-                            y_gorrito, raw_img = basic_predict(selected_models, img)
-                        finally:
-                            st.write(y_gorrito, final_weights)
+                            y_gorrito, raw_img = basic_predict(selected_models, final_weights, img)
         
                 if round(float(y_gorrito*100)) >= threshold:
                     st.success("¡Patacón Detectado!")