Update app.py

app.py

@@ -84,7 +84,7 @@ def process_text(prompt):
 @spaces.GPU
 def generate(
     prompt: str,
-    size_option: str = "竖版",
+    # size_option: str = "竖版",
     negative_prompt: str = "",
     prompt_2: str = "",
     negative_prompt_2: str = "",
@@ -110,14 +110,14 @@ def generate(
         negative_prompt_2 = None  # type: ignore
 
     
-    if size_option == "横版":
-        width, height = 1024, 736
-    elif size_option == "竖版":
-        width, height = 736, 1024
-    elif size_option == "方形":
-        width, height = 736, 736
-    else:
-        width, height = 736, 1024  # 可以定义一个默认值
+    # if size_option == "横版":
+    #     width, height = 1024, 736
+    # elif size_option == "竖版":
+    #     width, height = 736, 1024
+    # elif size_option == "方形":
+    #     width, height = 736, 736
+    # else:
+    #     width, height = 736, 1024  # 可以定义一个默认值
         
     # process_text("里面做一个测试")
     # print("prompt是：", prompt)
@@ -188,8 +188,8 @@ with gr.Blocks(css="style.css") as demo:
             run_button = gr.Button("生成", scale=0)
         result = gr.Image(label="生成结果", show_label=False)
 
-        # 使用 Radio 组件替代两个 Slider 组件
-        size_option = gr.Radio(choices=["横版", "竖版", "方形"], label="选择尺寸", value="竖版")
+        # # 使用 Radio 组件替代两个 Slider 组件
+        # size_option = gr.Radio(choices=["横版", "竖版", "方形"], label="选择尺寸", value="竖版")
         
     with gr.Accordion("高级选项", open=False):
         with gr.Row():
@@ -224,21 +224,21 @@ with gr.Blocks(css="style.css") as demo:
         )
         randomize_seed = gr.Checkbox(label="随机种子数", value=True)
         
-        # with gr.Row():
-        #     width = gr.Slider(
-        #         label="宽度",
-        #         minimum=256,
-        #         maximum=MAX_IMAGE_SIZE,
-        #         step=32,
-        #         value=736,
-        #     )
-        #     height = gr.Slider(
-        #         label="高度",
-        #         minimum=256,
-        #         maximum=MAX_IMAGE_SIZE,
-        #         step=32,
-        #         value=1024,
-        #     )
+        with gr.Row():
+            width = gr.Slider(
+                label="宽度",
+                minimum=256,
+                maximum=MAX_IMAGE_SIZE,
+                step=32,
+                value=736,
+            )
+            height = gr.Slider(
+                label="高度",
+                minimum=256,
+                maximum=MAX_IMAGE_SIZE,
+                step=32,
+                value=1024,
+            )
         
         apply_refiner = gr.Checkbox(label="增加精炼模型（refiner）", value=False, visible=ENABLE_REFINER)
         
@@ -328,7 +328,7 @@ with gr.Blocks(css="style.css") as demo:
         fn=generate,
         inputs=[
             prompt,
-            size_option,
+            # size_option,
             negative_prompt,
             prompt_2,
             negative_prompt_2,
@@ -336,8 +336,8 @@ with gr.Blocks(css="style.css") as demo:
             use_prompt_2,
             use_negative_prompt_2,
             seed,
-            # width,
-            # height,
+            width,
+            height,
             guidance_scale_base,
             guidance_scale_refiner,
             num_inference_steps_base,