Update app.py

app.py

@@ -92,7 +92,6 @@ def generate_flux_image(
     height: int,
     guidance_scale: float,
     num_inference_steps: int,
-    lora_scale: float,
     progress: gr.Progress = gr.Progress(track_tqdm=True),
 ) -> Image.Image:
     """Generate image using Flux pipeline"""
@@ -107,7 +106,6 @@ def generate_flux_image(
         width=width,
         height=height,
         generator=generator,
-        joint_attention_kwargs={"scale": lora_scale},
     ).images[0]
     
     # Save the generated image
@@ -199,7 +197,6 @@ with gr.Blocks() as demo:
                 with gr.Row():
                     guidance_scale = gr.Slider(0.0, 10.0, label="Guidance Scale", value=3.5, step=0.1)
                     num_inference_steps = gr.Slider(1, 50, label="Steps", value=30, step=1)
-                    lora_scale = gr.Slider(0.0, 1.0, label="LoRA Scale", value=1.0, step=0.1)
 
             with gr.Accordion("3D Generation Settings", open=False):
                 gr.Markdown("Stage 1: Sparse Structure Generation")
@@ -241,7 +238,7 @@ with gr.Blocks() as demo:
 
     generate_btn.click(
         generate_flux_image,
-        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, lora_scale],
+        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs=[generated_image],
     ).then(
         image_to_3d,