Update app.py

app.py

@@ -4,92 +4,17 @@ import io
 from PIL import Image
 import os
 
-# Stable Diffusion XL API 設定
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-import gradio as gr
-import requests
-import io
-from PIL import Image
-import os
-
-# Stable Diffusion XL API 設定
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-import gradio as gr
-import requests
-import io
-from PIL import Image
-import os
-
 # Stable Diffusion XL API 設定
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 headers = {"Authorization": "Bearer hf_MySpaceToken"}
 
-def generate_image(prompt):
-    # 呼叫 API 生成圖片
-    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-    
-    # 將回應轉換為圖片
-    image = Image.open(io.BytesIO(response.content))
-    
-    return image
-
-# 創建 Gradio 介面
-demo = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(
-        label="輸入圖片描述", 
-        placeholder="例如: a beautiful sunset over mountains, realistic photo"
-    ),
-    outputs=gr.Image(label="生成的圖片"),
-    title="Stable Diffusion XL 圖像生成器",
-    description="使用 Stable Diffusion XL 模型生成高品質圖像",
-    examples=[
-        ["a beautiful sunset over mountains, realistic photo"],
-        ["a cute cat playing with yarn, digital art"],
-        ["a futuristic city at night with neon lights"],
-        ["a magical forest with glowing mushrooms, fantasy art"]
-    ]
-)
-
-demo.launch()
-
-def generate_image(prompt):
-    # 呼叫 API 生成圖片
-    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-    
-    # 將回應轉換為圖片
-    image = Image.open(io.BytesIO(response.content))
-    
-    return image
-
-# 創建 Gradio 介面
-demo = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(
-        label="輸入圖片描述", 
-        placeholder="例如: a beautiful sunset over mountains, realistic photo"
-    ),
-    outputs=gr.Image(label="生成的圖片"),
-    title="Stable Diffusion XL 圖像生成器",
-    description="使用 Stable Diffusion XL 模型生成高品質圖像",
-    examples=[
-        ["a beautiful sunset over mountains, realistic photo"],
-        ["a cute cat playing with yarn, digital art"],
-        ["a futuristic city at night with neon lights"],
-        ["a magical forest with glowing mushrooms, fantasy art"]
-    ]
-)
-
-demo.launch()
 
 def generate_image(prompt):
     # 呼叫 API 生成圖片
     response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
     
-    # 將回應轉換為圖片
-    image = Image.open(io.BytesIO(response.content))
-    
-    return image
+    # 回傳圖片
+    return response.content
 
 # 創建 Gradio 介面
 demo = gr.Interface(
@@ -98,7 +23,7 @@ demo = gr.Interface(
         label="輸入圖片描述", 
         placeholder="例如: a beautiful sunset over mountains, realistic photo"
     ),
-    outputs=gr.Image(label="生成的圖片"),
+    outputs=gr.Image(type="bytes"),
     title="Stable Diffusion XL 圖像生成器",
     description="使用 Stable Diffusion XL 模型生成高品質圖像",
     examples=[