Update app.py

app.py

@@ -81,16 +81,16 @@ if __name__ == "__main__":
 
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 
-    # model = instantiate_from_config(config.model)
-    # model.load_state_dict(torch.load(opt.ckpt, map_location="cpu")["state_dict"], strict=False)
-    # model.cuda()
-    # model.eval()
-    # model = model.to(device)
-    # sampler = DDIMSampler(model)
+    model = instantiate_from_config(config.model)
+    model.load_state_dict(torch.load(opt.ckpt, map_location="cpu")["state_dict"], strict=False)
+    model.cuda()
+    model.eval()
+    model = model.to(device)
+    sampler = DDIMSampler(model)
 
     precision_scope = autocast
 
-    # @spaces.GPU
+    @spaces.GPU
     def start_tryon(human_img,garm_img):
         #load human image
         human_img = human_img['background'].convert("RGB").resize((768,1024))