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=1):
+def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_scale=7.0,model="Real6.0",num_images=1, width, height):
         
     if model == "Real5.0":
         model_id = "SG161222/Realistic_Vision_V5.0_noVAE"
@@ -41,8 +41,8 @@ def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_sca
         cross_attention_kwargs = {"scale":1},
         num_inference_steps = num_inference_steps,
         guidance_scale = guidance_scale,
-        width = 720,
-        height = 720,
+        width = width,
+        height = height,
         num_images_per_prompt=num_images
     )
     
@@ -68,11 +68,13 @@ with gr.Blocks() as demo:
             )
             generate_button = gr.Button("Generate Image")
             with gr.Accordion("Advanced Options", open=False):
+                
                 model = gr.Dropdown(
                     choices=["Real6.0","Real5.1","Real5.0"],
                     value="Real6.0",
                     label="Model",
                 )
+                
                 num_images = gr.Slider(  # New slider for number of images
                         minimum=1,
                         maximum=4,
@@ -80,6 +82,20 @@ with gr.Blocks() as demo:
                         step=1,
                         label="Number of Images to Generate"
                 )
+                width = gr.Slider(
+                        minimum=256,
+                        maximum=1024,
+                        value=512,
+                        step=64,
+                        label="Image Width"
+                )
+                height = gr.Slider(
+                    minimum=256,
+                    maximum=1024,
+                    value=512,
+                    step=64,
+                    label="Image Height"
+                )
                 steps_slider = gr.Slider(
                     minimum=1,
                     maximum=100,
@@ -108,7 +124,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],
+        inputs=[prompt, negative_prompt, steps_slider, guidance_slider, model, num_images, width, height],
         outputs=gallery
     )