Update app.py

app.py

@@ -233,6 +233,7 @@ def generate_image(
     adapter_strength_ratio,
     guidance_scale,
     seed,
+    max_side,
     progress=gr.Progress(track_tqdm=True),
 ):
 
@@ -248,7 +249,7 @@ 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=1280)
+    face_image = resize_img(face_image, max_side=max_side)
     # face_image = resize_img(face_image)
     face_image_cv2 = convert_from_image_to_cv2(face_image)
     height, width, _ = face_image_cv2.shape
@@ -270,7 +271,7 @@ def generate_image(
     face_kps = draw_kps(convert_from_cv2_to_image(face_image_cv2), face_info["kps"])
     if pose_image_path is not None:
         pose_image = load_image(pose_image_path)
-        pose_image = resize_img(pose_image, max_side=1280)
+        pose_image = resize_img(pose_image, max_side=max_side)
         # pose_image = resize_img(pose_image)
         pose_image_cv2 = convert_from_image_to_cv2(pose_image)
 
@@ -389,6 +390,13 @@ with gr.Blocks(css=css) as demo:
                     step=1,
                     value=42,
                 )
+                max_side = gr.Slider(
+                    label="Max side",
+                    minimum=512,
+                    maximum=2048,
+                    step=64,
+                    value=1280,
+                )
 
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
@@ -417,6 +425,7 @@ with gr.Blocks(css=css) as demo:
                 adapter_strength_ratio,
                 guidance_scale,
                 seed,
+                max_side,
             ],
             outputs=[gallery, usage_tips],
         )