Update app.py

app.py

@@ -248,8 +248,8 @@ def generate_image(
     # prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
 
     face_image = load_image(face_image_path)
-    # face_image = resize_img(face_image, max_side=1024)
-    face_image = resize_img(face_image)
+    face_image = resize_img(face_image, max_side=1280)
+    # face_image = resize_img(face_image)
     face_image_cv2 = convert_from_image_to_cv2(face_image)
     height, width, _ = face_image_cv2.shape
 
@@ -271,17 +271,15 @@ def generate_image(
     img_controlnet = face_image
     if pose_image_path is not None:
         pose_image = load_image(pose_image_path)
-        # pose_image = resize_img(pose_image, max_side=1024)
-        pose_image = resize_img(pose_image)
+        pose_image = resize_img(pose_image, max_side=1280)
+        # pose_image = resize_img(pose_image)
         img_controlnet = pose_image
         pose_image_cv2 = convert_from_image_to_cv2(pose_image)
 
         face_info = app.get(pose_image_cv2)
 
         if len(face_info) == 0:
-            raise gr.Error(
-                f"Cannot find any face in the reference image! Please upload another person image"
-            )
+            raise gr.Error(f"Cannot find any face in the reference image! Please upload another person image")
 
         face_info = face_info[-1]
         face_kps = draw_kps(pose_image, face_info["kps"])