Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 import numpy as np
 from PIL import Image
 import gradio as gr
+from io import BytesIO
 
 
 def gen_image(desc: str):
@@ -15,7 +16,8 @@ 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/dce9ad51-7cde-4eeb-8291-ae29f267ed2c/wukong-hf"
+    url = f"https://a2f051d4cabf45f885d7b0108edc9b9c.infer.ovaijisuan.com/v1/infers/975eedfd-6e15-4571-8ca9-b945da0da24b/wukong-hf"
     body = {
         "desc": desc,
         "style": ""
@@ -24,8 +26,11 @@ def gen_image(desc: str):
     resp_data = requests.post(url, json=body, headers=headers)
     if resp_data['status'] != 200:
         return []
-    img_np = resp_data['output_image_list'][0]
-    image = Image.fromarray(np.uint8(img_np))
+    # 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))
+
     return [image]