Update pages/Entorno de Ejecución.py

pages/Entorno de Ejecución.py

@@ -34,17 +34,11 @@ DIR = os.path.join(root_dir, "models")
 models = os.listdir(DIR)
 
 model_dict = dict()
-#for model in models:
-#    model_name = model.split(DIR)
-#    model_dict[model_name] = 0
-#model_list = []
-
 for model in models:
     model_name = model.split(DIR)
     model_name = str(model.split('.h5')[0])
     model_dir = os.path.join(DIR, model)
     model_dict[model_name] = model_dir
-    #model_list.append(os.path.join(DIR, model))
 
 def extract_number(filename):
     match = re.search(r'\d+', filename)
@@ -63,10 +57,10 @@ def sort_key(filename):
 # sort the filenames using the custom sorting key
 sorted_keys = sorted(model_dict.keys(), key=sort_key)
 
-model_dict = {k: model_dict[k] for k in sorted_keys}
+sorted_model_dict = {k: model_dict[k] for k in sorted_keys}
 
 # Create a dropdown menu to select the model
-model_choice = st.multiselect("Seleccione un modelo de clasificación", model_dict.keys())
+model_choice = st.multiselect("Seleccione un modelo de clasificación", sorted_model_dict.keys())
 
 selected_models = []
 for model in model_choice: