Update app.py

app.py

@@ -20,7 +20,7 @@ translator = Translator()
 
 
 
-def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
+def randomize_seed_fn(seed: int, randomize_seed: bool) -> float:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
@@ -82,11 +82,9 @@ def apply_style(style_name: str, positive: str) -> Tuple[str]:
 
 def generate(
     prompt: str,
-    seed: int = 0,
-    width: int = 1024,
-    height: int = 1024,
+    seed: float = 0,
+    steps: float = 1,
     style: str = DEFAULT_STYLE_NAME,
-    num_images: int = 2,
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -96,15 +94,14 @@ def generate(
     # print(client.view_api())
 
     prompt = apply_style(style, prompt)
-
+    strength = 0.0
     result = client.predict(
-		prompt=prompt,
-		seed=seed,
-		height=height,
-		width=width,
-		num_images=num_images,
-		fast_vae_decode=True,
-		api_name="/inference"
+        None,
+		prompt,
+		seed,
+        strength,
+        steps,
+		api_name="/predict"
     )
     images = result[0]
     print(images)
@@ -147,12 +144,12 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
         result = gr.Gallery(label="作品", columns=1, preview=True)
     with gr.Accordion("高级选项", open=False):
         with gr.Row():
-            num_images = gr.Slider(
-                label="数量",
+            steps = gr.Slider(
+                label="步数",
                 minimum=1,
-                maximum=6,
+                maximum=10,
                 step=1,
-                value=2,
+                value=1,
             )
             seed = gr.Slider(
                 label="种子",
@@ -163,6 +160,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
                 visible=True
             )
             randomize_seed = gr.Checkbox(label="随机种子", value=True)
+'''
         with gr.Row(visible=True):
             width = gr.Slider(
                 label="宽",
@@ -178,6 +176,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
                 step=8,
                 value=1024,
             )
+'''
     with gr.Row(visible=True):
         style_selection = gr.Radio(
             show_label=True,
@@ -209,10 +208,8 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
         inputs=[
             prompt,
             seed,
-            width,
-            height,
+            steps,
             style_selection,
-            num_images,
             randomize_seed,
         ],
         outputs=[result, seed],