Update app.py

app.py

@@ -213,7 +213,7 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
                 del lora_model
                 gc.collect()
 
-def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_strength, image_strength, sdxl_loras, sdxl_loras_new, progress=gr.Progress(track_tqdm=True)):
+def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, sdxl_loras, sdxl_loras_new, progress=gr.Progress(track_tqdm=True)):
     global last_lora, last_merged, last_fused, pipe
     face_info = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
     face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
@@ -276,7 +276,7 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
         height=1024,
         image_embeds=face_emb,
         image=face_image,
-        strength=image_strength,
+        strength=1-image_strength,
         control_image=images,
         num_inference_steps=20,
         guidance_scale = 7,
@@ -339,12 +339,13 @@ with gr.Blocks(css="custom.css") as demo:
             result = gr.Image(
                 interactive=False, label="Generated Image", elem_id="result-image"
             )
-            face_strength = gr.Slider(0, 1, value=0.85, step=0.1, label="Face strength", info="Higher values increase the face likeness but reduce the creative liberty of the models")
-            image_strength = gr.Slider(0, 1, value=0.85, step=0.1, label="Image strength", info="Higher values increase the similarity with the structure of the original photo")
+            face_strength = gr.Slider(0, 1, value=0.85, step=0.01, label="Face strength", info="Higher values increase the face likeness but reduce the creative liberty of the models")
+            image_strength = gr.Slider(0, 1, value=0.15, step=0.01, label="Image strength", info="Higher values increase the similarity with the structure/colors of the original photo")
             with gr.Accordion("Advanced options", open=False):
                 negative = gr.Textbox(label="Negative Prompt")
                 weight = gr.Slider(0, 10, value=0.9, step=0.1, label="LoRA weight")
-                
+                guidance_scale = gr.Slider(0, 50, value=7, step=0.1, label="Guidance Scale")
+                depth_control_scale = gr.Slider(0, 1, value=0.8, step=0.01, label="Zoe Depth ControlNet strenght")
     with gr.Column(elem_id="extra_info"):
         with gr.Accordion(
             "Use it with: 🧨  diffusers, ComfyUI, Invoke AI, SD.Next, AUTO1111",
@@ -409,7 +410,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, gr_sdxl_loras, gr_sdxl_loras_new],
+        inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
         outputs=[result, share_group],
     )
     button.click(
@@ -419,7 +420,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, gr_sdxl_loras, gr_sdxl_loras_new],
+        inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, gr_sdxl_loras, gr_sdxl_loras_new],
         outputs=[result, share_group],
     )
     share_button.click(None, [], [], js=share_js)