Update app.py

app.py

@@ -185,6 +185,7 @@ def generate_30(
     height: int = 768,
     guidance_scale: float = 4,
     num_inference_steps: int = 125,
+    txt_strength: float = 1.0,
     use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
@@ -282,6 +283,9 @@ def generate_30(
     prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
     print('catted combined meaned shape: ', prompt_embeds.shape)
 
+    prompt_embeds = prompt_embeds * txt_strength
+    pooled_prompt_embeds = pooled_prompt_embeds * txt_strength
+    
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
@@ -320,6 +324,7 @@ def generate_60(
     height: int = 768,
     guidance_scale: float = 4,
     num_inference_steps: int = 125,
+    txt_strength: float = 1.0,
     use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
@@ -417,6 +422,9 @@ def generate_60(
     prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
     print('catted combined meaned shape: ', prompt_embeds.shape)
 
+    prompt_embeds = prompt_embeds * txt_strength
+    pooled_prompt_embeds = pooled_prompt_embeds * txt_strength
+    
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
@@ -455,6 +463,7 @@ def generate_90(
     height: int = 768,
     guidance_scale: float = 4,
     num_inference_steps: int = 125,
+    txt_strength: float = 1.0,
     use_resolution_binning: bool = True, 
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
@@ -552,6 +561,9 @@ def generate_90(
     prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
     print('catted combined meaned shape: ', prompt_embeds.shape)
 
+    prompt_embeds = prompt_embeds * txt_strength
+    pooled_prompt_embeds = pooled_prompt_embeds * txt_strength
+    
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
@@ -650,6 +662,13 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
                     value="('deformed', 'distorted', 'disfigured':1.3),'not photorealistic':1.5, 'poorly drawn', 'bad anatomy', 'wrong anatomy', 'extra limb', 'missing limb', 'floating limbs', 'poorly drawn hands', 'poorly drawn feet', 'poorly drawn face':1.3, 'out of frame', 'extra limbs', 'bad anatomy', 'bad art', 'beginner',  'distorted face','amateur'",
                     visible=True,
                 )
+                text_strength = gr.Slider(
+                    label="Text Strength",
+                    minimum=0.0,
+                    maximum=5.0,
+                    step=0.01,
+                    value=1.0,
+                )
         with gr.Row():
             width = gr.Slider(
                 label="Width",
@@ -710,6 +729,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             height,
             guidance_scale,
             num_inference_steps,
+            text_strength,
         ],
         outputs=[result],
     )
@@ -730,6 +750,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             height,
             guidance_scale,
             num_inference_steps,
+            text_strength,
         ],
         outputs=[result],
     )
@@ -750,6 +771,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             height,
             guidance_scale,
             num_inference_steps,
+            text_strength,
         ],
         outputs=[result],
     )