Update app.py

app.py

@@ -88,10 +88,10 @@ def infer(
     if model_id != model_id_default:
         pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
         pipe = pipe.to(device)
-        pipe.fuse_lora(lora_scale=0.4)
+        pipe.fuse_lora(lora_scale=lora_scale)
         image = pipe(**params).images[0]
     else:
-        pipe_default.fuse_lora(lora_scale=0.4)
+        pipe_default.fuse_lora(lora_scale=lora_scale)
         image = pipe_default(**params).images[0]
 
     return image
@@ -171,7 +171,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=1024,
+                    value=512,
                 )
             
             with gr.Row():
@@ -180,7 +180,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=1024,
+                    value=512,
                 )
 
         run_button = gr.Button("Run", scale=1, variant="primary")