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, 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, 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,11 +276,11 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, sdxl_lora
         height=1024,
         image_embeds=face_emb,
         image=face_image,
-        strength=0.85,
+        strength=image_strength,
         control_image=images,
         num_inference_steps=20,
         guidance_scale = 7,
-        controlnet_conditioning_scale=[0.8, 0.8],
+        controlnet_conditioning_scale=[face_strength, 0.8],
     ).images[0]
     last_lora = repo_name
     gc.collect()
@@ -339,9 +339,12 @@ 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")
             with gr.Accordion("Advanced options", open=False):
                 negative = gr.Textbox(label="Negative Prompt")
-                weight = gr.Slider(0, 10, value=0.8, step=0.1, label="LoRA weight")
+                weight = gr.Slider(0, 10, value=0.9, step=0.1, label="LoRA weight")
+                
     with gr.Column(elem_id="extra_info"):
         with gr.Accordion(
             "Use it with: 🧨  diffusers, ComfyUI, Invoke AI, SD.Next, AUTO1111",
@@ -406,7 +409,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[photo, prompt, negative, weight, selected_state, gr_sdxl_loras, gr_sdxl_loras_new],
+        inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, gr_sdxl_loras, gr_sdxl_loras_new],
         outputs=[result, share_group],
     )
     button.click(
@@ -416,7 +419,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[photo, prompt, negative, weight, selected_state, gr_sdxl_loras, gr_sdxl_loras_new],
+        inputs=[photo, prompt, negative, weight, selected_state, face_strength, image_strength, gr_sdxl_loras, gr_sdxl_loras_new],
         outputs=[result, share_group],
     )
     share_button.click(None, [], [], js=share_js)