Update app.py

app.py

@@ -22,7 +22,7 @@ pipe.fuse_lora()
 MAX_SEED = 2**32-1
 
 @spaces.GPU(duration=75)
-def generate_image(prompt, steps=28, seed=None, cfg_scale=3.5, width=1024, height=1024, lora_scale=0.95):
+def generate_image(prompt, steps=28, seed=None, cfg_scale=2.8, width=1024, height=1024, lora_scale=0.5):
     if seed is None:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
@@ -38,16 +38,16 @@ def generate_image(prompt, steps=28, seed=None, cfg_scale=3.5, width=1024, heigh
     ).images[0]
     return image
 
-def run_lora(prompt, cfg_scale=3.5, steps=28, randomize_seed=True, seed=None, width=1024, height=1024, lora_scale=0.95):
+def run_lora(prompt, cfg_scale=2.8, steps=28, randomize_seed=True, seed=None, width=1024, height=1024, lora_scale=0.5):
     # Handle the case when only prompt is provided (for Examples)
     if isinstance(prompt, str) and all(param is None for param in [cfg_scale, steps, randomize_seed, seed, width, height, lora_scale]):
-        cfg_scale = 3.5
+        cfg_scale = 2.8
         steps = 28
         randomize_seed = True
         seed = None
         width = 1024
         height = 1024
-        lora_scale = 0.95
+        lora_scale = 0.5
 
     if randomize_seed or seed is None:
         seed = random.randint(0, MAX_SEED)
@@ -75,10 +75,10 @@ custom_css = """
 title = """<h1 align="center">FLUX Creativity LoRA</h1>
 """
 examples = [
-    "anime, cartoon, Hyper-detailed, endearing anime girl, bathed in a vibrant, colorful psychedelic glow, wearing dazzling, holographic Liquid Metal outfit, in a cozy tatami studio",
-    "extraterrestrial visage, close-up, highly intricate, ultra-detailed, full high definition",
-    "a full body photo shot of a beautiful and breathtaking image of a ((Man) ) wearing a fully clothed casual witchy witch clothes with intricate details in the style of a reapers cloak, he is holding a long curved double edged ((scythe) ). This full body image is a one of a kind unique highly detailed with 8k sharp focus quality masterpiece, hyper detailed, extremely detailed",
-    "schizophrenia attacks,go haywire, go crazy, hyper detailed, extremely detailed",
+    ["anime, cartoon, Hyper-detailed, endearing anime girl, bathed in a vibrant, colorful psychedelic glow, wearing dazzling, holographic Liquid Metal outfit, in a cozy tatami studio", 0.5], 
+    ["extraterrestrial visage, close-up, highly intricate, ultra-detailed, full high definition", 0.5],
+    ["a full body photo shot of a beautiful and breathtaking image of a ((Man) ) wearing a fully clothed casual witchy witch clothes with intricate details in the style of a reapers cloak, he is holding a long curved double edged ((scythe) ). This full body image is a one of a kind unique highly detailed with 8k sharp focus quality masterpiece, hyper detailed, extremely detailed", 0.5],
+    ["schizophrenia attacks,go haywire, go crazy, hyper detailed, extremely detailed", 0.5],
 ]
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray"), css=custom_css) as app:
@@ -110,7 +110,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="gray"), c
 
     gr.Examples(
         examples=examples,
-        inputs=[prompt],
+        inputs=[prompt, lora_scale],
         outputs=[result, seed],
         fn=run_lora,
         cache_examples=True