convert file type

app.py

@@ -5,6 +5,11 @@ import os
 import re
 import random
 from words import *
+from base64 import b64decode
+from PIL import Image
+import io
+import numpy as np
+
 
 # GPT-J-6B API
 API_URL = "https://api-inference.huggingface.co/models/EleutherAI/gpt-j-6B"
@@ -151,7 +156,14 @@ def desc_to_image_cf(desc):
   im_prompt_weight =1
   result = cf.fns[0].fn(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, range_scale, init_scale, seed, image_prompts,timestep_respacing, cutn, im_prompt_weight)
   
-  return result[0]
+
+  #convert result from dataurl to image
+  img=result[0]
+  header, encoded = img.split(",", 1)
+  data = b64decode(encoded)
+  image = Image.open(io.BytesIO(data))
+  image_np = np.array(image)
+  return image_np
 
 
 demo = gr.Blocks()