Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -204,6 +204,10 @@ def load_model():
|
|
204 |
authors.append(model_author)
|
205 |
model_path = f"/weights/{category_folder}/{character_name}/{model_name}"
|
206 |
import pickle
|
|
|
|
|
|
|
|
|
207 |
cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu", pickle_module=pickle)
|
208 |
model_version = cpt.get("version", "v1")
|
209 |
print(f"Indexed model {model_title} by {model_author} ({model_version})")
|
|
|
204 |
authors.append(model_author)
|
205 |
model_path = f"/weights/{category_folder}/{character_name}/{model_name}"
|
206 |
import pickle
|
207 |
+
path = f"/weights/{category_folder}/{character_name}/{model_name}"
|
208 |
+
|
209 |
+
if not os.path.exists(path):
|
210 |
+
print(f"Model file not found: {path}")
|
211 |
cpt = torch.load(f"/weights/{category_folder}/{character_name}/{model_name}", map_location="cpu", pickle_module=pickle)
|
212 |
model_version = cpt.get("version", "v1")
|
213 |
print(f"Indexed model {model_title} by {model_author} ({model_version})")
|