Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 import requests
 import os
 
-# 從環境變數讀取 Hugging Face API token
-API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/ukiyo-e-art"
-HF_API_TOKEN = os.getenv("HF_API_TOKEN")  # 確認您已將此變數在 Hugging Face Spaces 的 Secret 中設定
+# 使用 GPT-2 模型測試
+API_URL = "https://api-inference.huggingface.co/models/gpt2"
+HF_API_TOKEN = os.getenv("HF_API_TOKEN")  # 確保您在 Secrets 中設置了 API token
 headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
 def query(payload):
@@ -22,15 +22,15 @@ def inference(text):
         return f"Error: {output['error']}"
     
     # 返回模型的輸出結果
-    return output  # 根據實際返回的格式處理
+    return output
 
 # 創建 Gradio 介面
 interface = gr.Interface(
     fn=inference,
-    inputs=gr.Textbox(lines=2, placeholder="輸入生成藝術的描述..."),
-    outputs="text",  # 如果模型輸出的是圖片，可以改為 `outputs="image"`
-    title="Ukiyo-e Art 模型",
-    description="這是一個呼叫 Hugging Face 上 Ukiyo-e Art 模型的應用"
+    inputs=gr.Textbox(lines=2, placeholder="輸入文本..."),
+    outputs="text",
+    title="GPT-2 模型測試",
+    description="測試 Hugging Face API token 與 GPT-2 模型"
 )
 
 # 啟動 Gradio 應用