Update app.py

app.py

@@ -9,13 +9,14 @@ from typing import Tuple
 from translatepy import Translator
 
 MODEL = os.environ.get("MODEL")
+
 DESCRIPTION = """
 # IMGEN🧚‍♀️
-<center>🧚‍♂️输入描述生成图片，选择喜欢的风格~<br>🧚高级选项可以设置生成数量和大小等</center>
+<center>🧚‍♂️输入描述生成图片，选择喜欢的风格~<br>🧚默认使用Turbo快速生成，高级选项可更改。</center>
 """
 MODEL_ID="stabilityai/sdxl-turbo"
-MODEL_NAME = MODEL_ID.split("/")[-1]
-LICENSE = "采用 " + MODEL_NAME +" 模型"
+
+LICENSE = "采用 " + SDXL +" 模型"
 translator = Translator()
 
 
@@ -28,7 +29,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 MAX_SEED = np.iinfo(np.int32).max
 
 client = Client(MODEL)
-client_test = Client("https://google-sdxl.hf.space/")
+client2 = Client(MODEL2)
 
 style_list = [
     {
@@ -86,25 +87,38 @@ def generate(
     steps: float = 1,
     style: str = DEFAULT_STYLE_NAME,
     randomize_seed: bool = False,
+    use_turbo: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
     prompt = str(translator.translate(prompt, 'English'))
     print(f'prompt:{prompt}')
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    print(client_test.view_api())
+    # print(client_test.view_api())
 
     prompt = apply_style(style, prompt)
     strength = 0.7
-    result = client.predict(
-        None,
-		prompt,
-		seed,
-        strength,
-        steps,
-		fn_index=1
-    )
-    print(result)
-    images = MODEL+"file="+result
+    if use_turbo:
+        result = client.predict(
+            None,
+    		prompt,
+    		seed,
+            strength,
+            steps,
+    		fn_index=1
+        )
+        print(result)
+        images = MODEL+"file="+result   
+    else:
+        seed = 0
+        result = client2.predict(
+            prompt,
+            None,
+            7.5,
+            "(No style)",
+            fn_index=3    
+        )
+        print(result)
+        images = result["data"][0]
 
     return images, seed
 
@@ -134,10 +148,10 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
                 max_lines=1,
                 placeholder="画什么好呢",
                 container=False,
-                scale=2,
+                scale=3,
             )
             run_button = gr.Button("生成", scale=1)
-        result = gr.Image(label="作品")
+        result = gr.Gallery(label="作品", columns = 1, preview=True)
     with gr.Accordion("高级选项", open=False):
         with gr.Row():
             steps = gr.Slider(
@@ -156,6 +170,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
                 visible=True
             )
             randomize_seed = gr.Checkbox(label="随机种子", value=True)
+            use_turbo = gr.Checkbox(label="快速生成", value=True)
     with gr.Row(visible=True):
         style_selection = gr.Radio(
             show_label=True,
@@ -189,6 +204,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
             steps,
             style_selection,
             randomize_seed,
+            use_turbo,
         ],
         outputs=[result, seed],
         api_name="run",