Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import io
+from PIL import Image
 
 # gr.Interface.load("models/runwayml/stable-diffusion-v1-5").launch()
 
@@ -9,7 +11,8 @@ headers = {"Authorization": "Bearer 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"}
 
 def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
-	return response.content
+    image = Image.open(io.BytesIO(response.content))
+	return image
 
 # 创建Gradio界面
 iface = gr.Interface(