| | import os |
| | import requests |
| | import numpy as np |
| | from PIL import Image |
| | import gradio as gr |
| | from io import BytesIO |
| |
|
| |
|
| | 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 |
| | """ |
| | if not desc: |
| | return |
| | access_token = os.environ['token'] |
| | headers = {'content-type': "application/json", 'X-Subject-Token': access_token} |
| |
|
| | |
| | url = "https://a2f051d4cabf45f885d7b0108edc9b9c.infer.ovaijisuan.com/v1/infers/975eedfd-6e15-4571-8ca9-b945da0da24b/wukong_hf" |
| | body = { |
| | "user_name": 'huggingface', |
| | "desc": desc, |
| | "style": style |
| | } |
| |
|
| | resp_data = requests.post(url, json=body, headers=headers) |
| | print(resp_data['status']) |
| | if resp_data['status'] != 200: |
| | return |
| | |
| | |
| | img_rep = requests.get(resp_data['output_image_url'][0]) |
| | image = Image.open(BytesIO(img_rep.content)) |
| | image_np = np.asarray(image) |
| |
|
| | return image_np |
| |
|
| |
|
| | demo = gr.Interface( |
| | fn=gen_image, |
| | inputs=["text", "text"], |
| | outputs=gr.outputs.Image(type='numpy'), |
| | ) |
| | demo.launch() |
