Update app.py

app.py

@@ -248,7 +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, max_side=1024)
+    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,8 @@ 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, max_side=1024)
+        pose_image = resize_img(pose_image)
         img_controlnet = pose_image
         pose_image_cv2 = convert_from_image_to_cv2(pose_image)
 
@@ -285,19 +287,20 @@ def generate_image(
         face_kps = draw_kps(pose_image, face_info["kps"])
 
         width, height = face_kps.size
-
+    print(width, height)
     print("Start inference...")
     print(f"[Debug] Prompt: {prompt}, \n[Debug] Neg Prompt: {negative_prompt}")
 
-    pipe.set_ip_adapter_scale(adapter_strength_ratio)
+    # pipe.set_ip_adapter_scale(adapter_strength_ratio)
     images = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
         image=face_kps,
         face_emb=face_emb,
-        controlnet_conditioning_scale=controlnet_conditioning_scale,
+        controlnet_conditioning_scale=float(controlnet_conditioning_scale),
+        ip_adapter_scale=float(adapter_strength_ratio)
         num_inference_steps=num_steps,
-        guidance_scale=guidance_scale,
+        guidance_scale=float(guidance_scale),
         height=height,
         width=width,
         generator=torch.Generator(device=device).manual_seed(seed),
@@ -319,6 +322,7 @@ tips = r"""
 css = """
 .gradio-container {width: 85% !important}
 """
+
 with gr.Blocks(css=css) as demo:
     # description
     gr.Markdown(title)