Update app.py

app.py

@@ -97,7 +97,7 @@ pipe.fuse_lora()
 pipe = pipe.to(device)
 # pipe.enable_xformers_memory_efficient_attention()
 
-generator = torch.Generator(device='cuda:0').manual_seed(123456)
+generator = torch.Generator(device='cuda').manual_seed(123456)
 
 vae = pipe.vae
 
@@ -120,8 +120,8 @@ def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance
         negative_prompt = None
 
     
-    init_image = dict["image"].convert("RGB")#.resize((1024, 1024))
-    mask = dict["mask"].convert("L") #.resize((1024, 1024))
+    init_image = dict["image"].convert("RGB").resize((1024, 1024))
+    mask = dict["mask"].convert("L").resize((1024, 1024))
 
     width, height = get_size(init_image)