Kit-Lemonfoot commited on
Commit
b43f45b
1 Parent(s): a6cfc88

Fixed Choco and Lui. I'm well aware images are broken, seems to be something on HF's side -_-

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -147,6 +147,8 @@ def load_model():
147
  model_author = info.get("author", None)
148
  model_cover = f"weights/{category_folder}/{character_name}/{info['cover']}"
149
  model_index = f"weights/{category_folder}/{character_name}/{info['feature_retrieval_library']}"
 
 
150
  model_path = f"weights/{category_folder}/{character_name}/{model_name}"
151
  cpt = torch.load(f"weights/{category_folder}/{character_name}/{model_name}", map_location="cpu")
152
  model_version = cpt.get("version", "v1")
 
147
  model_author = info.get("author", None)
148
  model_cover = f"weights/{category_folder}/{character_name}/{info['cover']}"
149
  model_index = f"weights/{category_folder}/{character_name}/{info['feature_retrieval_library']}"
150
+ if info['feature_retrieval_library'] == "None":
151
+ model_index = None
152
  model_path = f"weights/{category_folder}/{character_name}/{model_name}"
153
  cpt = torch.load(f"weights/{category_folder}/{character_name}/{model_name}", map_location="cpu")
154
  model_version = cpt.get("version", "v1")