| 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() |
