Update app.py

app.py

@@ -8,7 +8,7 @@ lora_path = "OedoSoldier/detail-tweaker-lora"
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to("cuda")
 
 @spaces.GPU
-def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_scale=7.0,model="Real6.0",num_images_slider=1):
+def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_scale=7.0,model="Real6.0",num_images=1):
         
     if model == "Real5.0":
         model_id = "SG161222/Realistic_Vision_V5.0_noVAE"
@@ -73,7 +73,7 @@ with gr.Blocks() as demo:
                     value="Real6.0",
                     label="Model",
                 )
-                num_images_slider = gr.Slider(  # New slider for number of images
+                num_images = gr.Slider(  # New slider for number of images
                         minimum=1,
                         maximum=4,
                         value=1,
@@ -108,7 +108,7 @@ with gr.Blocks() as demo:
     # Connect the interface to the generation function
     generate_button.click(
         fn=generate_image,
-        inputs=[prompt, negative_prompt, steps_slider, guidance_slider, model, num_images_slider],
+        inputs=[prompt, negative_prompt, steps_slider, guidance_slider, model, num_images],
         outputs=gallery
     )