Spaces:
Running
Running
Update pages/Entorno de Ejecución.py
Browse files
pages/Entorno de Ejecución.py
CHANGED
@@ -132,7 +132,7 @@ with cnn:
|
|
132 |
|
133 |
#else:
|
134 |
with st.spinner('Cargando predicción...'):
|
135 |
-
selected_models = [load_model(model_dict[
|
136 |
final_weights = weight_list if len(weight_list) >= 1 else [1 for i in range(len(selected_models))]
|
137 |
y_gorrito, raw_img = predict(selected_models, final_weights, img)
|
138 |
|
|
|
132 |
|
133 |
#else:
|
134 |
with st.spinner('Cargando predicción...'):
|
135 |
+
selected_models = [load_model(model_dict[model_name]) for model_name in model_choice if model_name not in selected_models]
|
136 |
final_weights = weight_list if len(weight_list) >= 1 else [1 for i in range(len(selected_models))]
|
137 |
y_gorrito, raw_img = predict(selected_models, final_weights, img)
|
138 |
|