Update app.py

app.py

@@ -195,15 +195,15 @@ def main(args):
     config = OmegaConf.load("configs/config_customnet.yaml") 
     model = instantiate_from_config(config.model)
 
-    # model_path='./customnet_v1.pt?download=true'
-    # if not os.path.exists(model_path):
-    #     os.system(f'wget https://huggingface.co/TencentARC/CustomNet/resolve/main/customnet_v1.pt?download=true -P .')
+    model_path='./customnet_v1.pt?download=true'
+    if not os.path.exists(model_path):
+        os.system(f'wget https://huggingface.co/TencentARC/CustomNet/resolve/main/customnet_v1.pt?download=true -P .')
 
-    # ckpt = torch.load(model_path, map_location="cpu")
-    # model.load_state_dict(ckpt)
-    # del ckpt
+    ckpt = torch.load(model_path, map_location="cpu")
+    model.load_state_dict(ckpt)
+    del ckpt
     model = model.to(device)
-    sampler = DDIMSampler(model, device=device)
+    sampler = None
 
     # load demo
     demo = gr.Blocks()