Update app.py

Added multiple flexibility

app.py

@@ -18,7 +18,6 @@ pipe = DiffusionPipeline.from_pretrained("sayakpaul/FLUX.1-merged", torch_dtype=
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-
 def inferee(prompt, seed=42, randomize_seed=True, width=400, height=400, guidance_scale=3.5, num_inference_steps=8):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -33,10 +32,21 @@ def inferee(prompt, seed=42, randomize_seed=True, width=400, height=400, guidanc
     ).images[0] 
     return image, seed
 
+prompt = gr.Text(
+    label="Prompt",
+    show_label=False,
+    max_lines=1,
+    placeholder="Enter your prompt",
+    container=False)
+
+run_button = gr.Button("Run", scale=0)
+
+result = gr.Image(label="Result", show_label=False)
+
 interface = gr.Interface(
     fn=inferee,
-    inputs=["text"],
-    outputs=["image", "number"],
+    inputs=[prompt],
+    outputs=[result]
 )
 
 interface.launch()