Update app.py

app.py

@@ -1,15 +1,43 @@
+import os
 import gradio as gr
 from utils.t2i import t2i_gen
 
+MAX_SEED = np.iinfo(np.int32).max
+MIN_IMAGE_SIZE = int(os.getenv("MIN_IMAGE_SIZE", "512"))
+MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
+
 with gr.Blocks(
     title="🪄 LayerDiffuse - Flux version ☔",
     theme="CultriX/gradio-theme"
 ) as demo:
+    gr.Markdown(
+        """
+        # 🪄 LayerDiffuse - Flux version ☔
+
+        A Flux version implementation of LayerDiffuse ([LayerDiffuse](https://github.com/lllyasviel/LayerDiffuse))
+        """
+    )
     prompt = gr.Text(
         label="Prompt",
         info="Your prompt here",
         placeholder="E.g: glass bottle, high quality"
     )
 
+    with gr.Row:
+        width = gr.Slider(
+            label="Width",
+            minimum=MIN_IMAGE_SIZE,
+            maximum=MAX_IMAGE_SIZE,
+            step=32,
+            value=1024,
+        )
+        height = gr.Slider(
+            label="Height",
+            minimum=MIN_IMAGE_SIZE,
+            maximum=MAX_IMAGE_SIZE,
+            step=32,
+            value=1024,
+        )
+
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(show_error=True)