Update app.py

app.py

@@ -25,13 +25,14 @@ transformer = FluxTransformer2DModel.from_pretrained(
     subfolder="transformer", 
     torch_dtype=torch.bfloat16
 )
+
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", 
     transformer=transformer, 
     torch_dtype=torch.bfloat16
 )
-ip_model = IPAdapter(pipe, image_encoder_path, ipadapter_path, device="cuda", num_tokens=128)
 
+ip_model = IPAdapter(pipe, image_encoder_path, ipadapter_path, device="cuda", num_tokens=128)
 
 def resize_img(image, max_size=1024):
     width, height = image.size
@@ -95,12 +96,19 @@ with gr.Blocks(css=css) as demo:
                     label="Input Image",
                     type="pil"
                 )
+                scale = gr.Slider(
+                    label="Image Scale",
+                    minimum=0.0,
+                    maximum=1.0,
+                    step=0.1,
+                    value=0.7,
+                )
                 prompt = gr.Text(
                     label="Prompt",
                     max_lines=1,
                     placeholder="Enter your prompt",
                 )
-                run_button = gr.Button("Process", variant="primary")
+                run_button = gr.Button("Generate", variant="primary")
             
             with gr.Column():
                 result = gr.Image(label="Result")
@@ -122,7 +130,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=960,
+                    value=1024,
                 )
                 
                 height = gr.Slider(
@@ -130,16 +138,8 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=1280,
+                    value=1024,
                 )
-            
-            scale = gr.Slider(
-                label="Scale",
-                minimum=0.0,
-                maximum=1.0,
-                step=0.1,
-                value=0.7,
-            )
     
     run_button.click(
         fn=process_image,