Update app.py

app.py

@@ -11,7 +11,11 @@ from translatepy import Translator
 MODEL = os.environ.get("MODEL")
 API_URL = "https://api-inference.huggingface.co/models/tianweiy/DMD2"
 DESCRIPTION = """
-# DMD2 文生图
+# IMGEN🧚‍♀️DMD2
+
+🧚‍♂️输入描述生成图片，选择喜欢的风格~
+
+🧚高级选项可以设置生成数量和大小等
 """
 translator = Translator()
 
@@ -80,8 +84,8 @@ def apply_style(style_name: str, positive: str) -> Tuple[str, str]:
 def generate(
     prompt: str,
     seed: int = 0,
-    width: int = 1024,
-    height: int = 1024,
+    width: int = 512,
+    height: int = 512,
     style: str = DEFAULT_STYLE_NAME,
     num_images: int = 2,
     randomize_seed: bool = False,
@@ -144,7 +148,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
             num_images = gr.Slider(
                 label="数量",
                 minimum=1,
-                maximum=5,
+                maximum=6,
                 step=1,
                 value=2,
             )
@@ -161,16 +165,16 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
             width = gr.Slider(
                 label="宽",
                 minimum=512,
-                maximum=2048,
+                maximum=1024,
                 step=8,
-                value=1024,
+                value=512,
             )
             height = gr.Slider(
                 label="高",
                 minimum=512,
-                maximum=2048,
+                maximum=1024,
                 step=8,
-                value=1024,
+                value=512,
             )
     with gr.Row(visible=True):
             style_selection = gr.Radio(