Update app.py

app.py

@@ -121,16 +121,15 @@ def plex(mput, prompt, neg_prompt, stips, modal_id, dula, blip, blop):
     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
     generator = torch.Generator("cpu").manual_seed(random.randint(0, MAX_SEED))
     
-    tilage = pope(prompt,num_inference_steps=5).images[0]
-    cannmage = tilage
-    cannyimage = np.array(cannmage)
+    tilage = pope(prompt,num_inference_steps=5,height=512,width=512,generator=generator).images[0]
+    cannyimage = np.array(tilage)
     low_threshold = 100
     high_threshold = 200
     cannyimage = cv2.Canny(cannyimage, low_threshold, high_threshold)
     cannyimage = cannyimage[:, :, None]
     cannyimage = np.concatenate([cannyimage, cannyimage, cannyimage], axis=2)
     canny_image = Image.fromarray(cannyimage)  
-    pose_image = load_image(mput)
+    pose_image = load_image(mput).resize((512, 512))
     openpose_image = openpose(pose_image)
     images = [openpose_image, canny_image]