path

app.py

@@ -33,7 +33,7 @@ image_model_config_dict = OrderedDict({
         # "config": "./configs/image_cond_diffuser_asl/image-ASLDM-256.yaml",
         # "ckpt_path": "./checkpoints/image_cond_diffuser_asl/image-ASLDM-256.ckpt",
         "config": "/home/user/app/configs/image_cond_diffuser_asl/image-ASLDM-256.yaml",
-        "ckpt_path": "image_cond_diffuser_asl/image-ASLDM-256.ckpt",
+        "ckpt_path": "checkpoints/image_cond_diffuser_asl/image-ASLDM-256.ckpt",
     },
 })
 
@@ -42,11 +42,11 @@ text_model_config_dict = OrderedDict({
         # "config": "./configs/text_cond_diffuser_asl/text-ASLDM-256.yaml",
         # "ckpt_path": "./checkpoints/text_cond_diffuser_asl/text-ASLDM-256.ckpt",
         "config": "./configs/text_cond_diffuser_asl/text-ASLDM-256.yaml",
-        "ckpt_path": "text_cond_diffuser_asl/text-ASLDM-256.ckpt",
+        "ckpt_path": "checkpoints/text_cond_diffuser_asl/text-ASLDM-256.ckpt",
     },
 })
 
-model_path = snapshot_download(repo_id="Maikou/Michelangelo/tree/main/checkpoints")
+model_path = snapshot_download(repo_id="Maikou/Michelangelo")
 
 class InferenceModel(object):
     model = None