fix use_folds

app.py

@@ -98,9 +98,9 @@ comp_embeddings = get_comp_embeddings(emb_files, use_folds)
 # Preload embedding models, input sizes
 K.clear_session()
 embed_models, sizes = [], []
-for cfg_file, rst_file, npz_file in zip(cfg_files, rst_files, emb_files):
+for cfg_file, rst_file, use_fold in zip(cfg_files, rst_files, use_folds):
     cfg = get_cfg(cfg_file)
-    assert cfg.FOLD_TO_RUN == use_fold[npz_file]
+    assert cfg.FOLD_TO_RUN == use_fold
     cfg.pretrained = None  # avoid weight downloads
     if isinstance(cfg.IMAGE_SIZE, int):
         cfg.IMAGE_SIZE = (cfg.IMAGE_SIZE, cfg.IMAGE_SIZE)