Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 from io import BytesIO
 
 
-def gen_image(desc: str):
+def gen_image(desc: str, style: str):
     """generate the image from the wukong huahua model of ascend server in Wuhan AICC
     Args:
         desc(str): the input description text
@@ -19,8 +19,9 @@ def gen_image(desc: str):
     # url = f"https://a2f051d4cabf45f885d7b0108edc9b9c.infer.ovaijisuan.com/v1/infers/dce9ad51-7cde-4eeb-8291-ae29f267ed2c/wukong-hf"
     url = f"https://a2f051d4cabf45f885d7b0108edc9b9c.infer.ovaijisuan.com/v1/infers/975eedfd-6e15-4571-8ca9-b945da0da24b/wukong-hf"
     body = {
+        "user_name": 'huggingface',
         "desc": desc,
-        "style": ""
+        "style": style
     }
 
     resp_data = requests.post(url, json=body, headers=headers)