Update app.py

app.py

@@ -99,17 +99,18 @@ def generate(
     style: str = DEFAULT_STYLE_NAME,
     randomize_seed: bool = False,
     use_turbo: bool = True,
+    use_flash: bool = False,
     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
     images = []
-    if use_turbo:
+    if use_turbo and use_flash is False:
         result = client.predict(
             None,
     		prompt,
@@ -125,6 +126,20 @@ def generate(
         pil_image = Image.open(BytesIO(images_request.content))
         images.append(pil_image)
         print(f'B:{images}')
+    elif use_flash is True:
+        result = client_test.predict(
+            prompt,
+            0,
+            True,
+            api_name="/infer"
+        )
+        print(result)
+        image_url = MODEL+"file="+result
+        images_request = requests.get(image_url)
+        print(f'A:{images_request}')
+        pil_image = Image.open(BytesIO(images_request.content))
+        images.append(pil_image)
+        print(f'B:{images}')        
     else:
         seed = 0
         job = client2.submit(
@@ -193,6 +208,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
             )
             randomize_seed = gr.Checkbox(label="随机种子", value=True)
             use_turbo = gr.Checkbox(label="快速生成", value=True)
+            use_flash = gr.Checkbox(labe="Flash", value=False)
     with gr.Row(visible=True):
         style_selection = gr.Radio(
             show_label=True,
@@ -227,6 +243,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
             style_selection,
             randomize_seed,
             use_turbo,
+            use_flash,
         ],
         outputs=[result, seed],
         api_name="run",