Update app.py

app.py

@@ -1,35 +1,11 @@
-import gradio as gr
 import requests
-import io
-from PIL import Image
-from dotenv import load_dotenv
-import os
 
-# 載入環境變數
-load_dotenv()
-
-# Hugging Face API 設定
 API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/ukiyo-e-art"
-headers = {"Authorization": "Bearer hf_MySpaceToken"}
-
-def generate_image(prompt):
-    try:
-        # 呼叫 Hugging Face API
-        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-        # 轉換回應為圖像
-        image = Image.open(io.BytesIO(response.content))
-        return image
-    except Exception as e:
-        return None
-
-# 建立 Gradio 介面
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=gr.Textbox(label="請輸入圖像描述"),
-    outputs=gr.Image(label="生成的圖像"),
-    title="浮世繪風格圖像生成器",
-    description="輸入描述文字,生成浮世繪風格的圖像"
-)
-
-# 啟動應用
-iface.launch()
+headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+image_bytes = query({
+	"inputs": "Astronaut riding a horse",
+})