Update app.py

app.py

@@ -69,6 +69,7 @@ def process(
     randomize_seed_checkbox: bool,
     strength_slider: float,
     num_inference_steps_slider: int,
+    num_influence: float,
     progress=gr.Progress(track_tqdm=True)
 ):
     if not input_text:
@@ -99,7 +100,7 @@ def process(
         num_images_per_prompt =2,
         strength=strength_slider,
         generator=generator,
-        joint_attention_kwargs={"scale": 1.0},
+        joint_attention_kwargs={"scale": num_influence},
         num_inference_steps=num_inference_steps_slider
         )
     print('INFERENCE DONE')
@@ -142,6 +143,7 @@ with gr.Blocks() as demo:
                     placeholder="Enter your prompt",
                     container=False,
                 )
+                lorasteps = gr.Slider(label="Influence", minimum=0, maximum=2, step=0.1, value=1)
                 submit_button_component = gr.Button(
                     value='Submit', variant='primary', scale=0)
 
@@ -194,7 +196,8 @@ with gr.Blocks() as demo:
             seed_slicer_component,
             randomize_seed_checkbox_component,
             strength_slider_component,
-            num_inference_steps_slider_component
+            num_inference_steps_slider_component,
+            lorasteps
         ],
         outputs=[
             output_image_component,