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
@@ -16,57 +16,29 @@ def gen_image(desc: str):
     access_token = os.environ['token']
     headers = {'content-type': "application/json", 'X-Subject-Token': access_token}
 
-    # 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"
+    # url = "https://a2f051d4cabf45f885d7b0108edc9b9c.infer.ovaijisuan.com/v1/infers/dce9ad51-7cde-4eeb-8291-ae29f267ed2c/wukong_hf"
+    url = "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)
     if resp_data['status'] != 200:
-        return []
+        return
     # img_np = np.array(resp_data['output_image_list'][0])
     # image = Image.fromarray(np.uint8(img_np))
     img_rep = requests.get(resp_data['output_image_url'][0])
     image = Image.open(BytesIO(img_rep.content))
+    image_np = np.asarray(image)
 
-    return [image]
-
-
-examples = [
-    '天空之城 赛博朋克 动漫',
-    '秋水共长天一色',
-    '海滩 蓝天 美景',
-    '教堂 巴洛克风格',
-    '落日 莫奈',
-    '来自深渊 雪山飞狐'
-]
-
-block = gr.Blocks()
-
-
-with block:
-    with gr.Group():
-        with gr.Box():
-            with gr.Row().style(mobile_collapse=False, equal_height=True):
-                text = gr.Text(
-                    placeholder="输入中文，生成图片",
-                ).style(
-                    border=(True, False, True, True),
-                    rounded=(True, False, False, True),
-                    container=False,
-                )
-
-                btn = gr.Button("Generate image").style(
-                    margin=False,
-                    rounded=(False, True, True, False),
-                )
-
-        gallery = gr.Gallery(
-            label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[1, 1], height="auto")
+    return image_np
 
-        gr.Examples(examples=examples, fn=gen_image, inputs=text, outputs=gallery)
 
-block.queue(concurrency_count=3).launch(debug=True)
+demo = gr.Interface(
+    fn=gen_image,
+    inputs=["text", "text"],
+    outputs=gr.outputs.Image(type='numpy'),
+)
+demo.launch()