Update app.py

app.py

@@ -1,15 +1,25 @@
+import gradio as gr
 import requests
 
-API_URL = "https://api-inference.huggingface.co/models/KappaNeuro/ukiyo-e-art"
-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",
-})
-# You can access the image with PIL.Image for example
-import io
-from PIL import Image
-image = Image.open(io.BytesIO(image_bytes))
+# API调用函数
+def call_api(user_input):
+    url = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"  # 替换为你的API URL
+    headers = {"Authorization": "Bearer YOUR_TOKEN"}
+    params = {"query": user_input}
+
+    response = requests.get(url, headers=headers, params=params)
+    if response.status_code == 200:
+        return response.json()  # 返回API的响应
+    else:
+        return f"Error: {response.status_code}"
+
+# 使用Gradio界面
+iface = gr.Interface(
+    fn=call_api, 
+    inputs="text", 
+    outputs="json",
+    title="API 调用",
+    description="输入文本并从API获取结果"
+)
+
+iface.launch()