Spaces:
Running
Running
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
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")
|