update

app.py

@@ -174,11 +174,14 @@ t2i_default_img_path_list = []
 import base64, requests
 from io import BytesIO
 from PIL import Image
-def inference_image(text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale):
+def luck_inference_image(text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale):
+    return inference_image(text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale, luck=True)
+
+def inference_image(text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale, luck=False):
     try:
         res = requests.get(f"https://www.clueai.cn/clueai/hf_text2image?text={text}&negative_prompt={n_text}\
 &guidance_scale={guidance_scale}&num_inference_steps={steps}\
-&style={style}&shape={shape}&clarity={clarity}&shape_scale={shape_scale}")
+&style={style}&shape={shape}&clarity={clarity}&shape_scale={shape_scale}&luck={luck}")
     except Exception as e:
         logger.error(f"error, {e}")
         return 
@@ -255,7 +258,7 @@ with gr.Blocks(css=css, title="ClueAI") as demo:
         )
 
         input_params = [text, n_text, guidance_scale, style, shape, clarity, steps, shape_scale]
-        generate_prompt_btn.click(inference_image, inputs=input_params, outputs=[t2i_gallery])
+        generate_prompt_btn.click(luck_inference_image, inputs=input_params, outputs=[t2i_gallery])
         text.submit(inference_image, inputs=input_params, outputs=t2i_gallery)
         btn.click(inference_image, inputs=input_params, outputs=t2i_gallery)