Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -444,9 +444,9 @@ if __name__ == "__main__":
|
|
444 |
# Model configurations and loading
|
445 |
models = {}
|
446 |
model_paths = {
|
447 |
-
"upper": {"class_name": "
|
448 |
-
"lower": {"class_name": "
|
449 |
-
"dress": {"class_name": "
|
450 |
"multi": {"class_name": "TryOffDiffv2", "path": "tryoffdiffv2_multi.pth"}, # internal code: model_20250310_155608
|
451 |
}
|
452 |
for name, cfg in model_paths.items():
|
|
|
444 |
# Model configurations and loading
|
445 |
models = {}
|
446 |
model_paths = {
|
447 |
+
"upper": {"class_name": "TryOffDiffv2Single", "path": "tryoffdiffv2_upper.pth"}, # internal code: model_20250213_134430
|
448 |
+
"lower": {"class_name": "TryOffDiffv2Single", "path": "tryoffdiffv2_lower.pth"}, # internal code: model_20250213_134130
|
449 |
+
"dress": {"class_name": "TryOffDiffv2Single", "path": "tryoffdiffv2_dress.pth"}, # internal code: model_20250213_133554
|
450 |
"multi": {"class_name": "TryOffDiffv2", "path": "tryoffdiffv2_multi.pth"}, # internal code: model_20250310_155608
|
451 |
}
|
452 |
for name, cfg in model_paths.items():
|