Spaces:

frncscp
/

Patacotron

Sleeping

frncscp commited on Mar 18, 2023

Commit

373b5b1

1 Parent(s): bf4858a

Update pages/Entorno de Ejecución.py

Files changed (1) hide show

pages/Entorno de Ejecución.py CHANGED Viewed

@@ -69,7 +69,7 @@ with col_a:
     #for model in model_choice:
         #selected_models.append(model)
-    selected_models = [load_model(model) for model in model_choice]
     # Set the image dimensions
     IMAGE_WIDTH = IMAGE_HEIGHT = 224

     #for model in model_choice:
         #selected_models.append(model)
+    selected_models = [load_model(model_dict[model]) for model in model_choice]
     # Set the image dimensions
     IMAGE_WIDTH = IMAGE_HEIGHT = 224