Update app.py

app.py

@@ -4,6 +4,11 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+import os, io
+from PIL import Image
+from smolagents.agent_types import AgentImage
+from smolagents import tool
+
 
 from Gradio_UI import GradioUI
 
@@ -43,11 +48,7 @@ def text_to_image(prompt: str) -> AgentImage:
     if not hf_token:
         raise RuntimeError("HF_TOKEN is missing. Add it in Settings → Variables and Secrets.")
 
-    headers = {
-        "Authorization": f"Bearer {hf_token}",
-        "Accept": "image/png"
-    }
-    # 你也可以换成其它支持推理的模型，比如：black-forest-labs/FLUX.1-dev
+    headers = {"Authorization": f"Bearer {hf_token}", "Accept": "image/png"}
     url = "https://api-inference.huggingface.co/models/stabilityai/sdxl-turbo"
 
     resp = requests.post(url, headers=headers, json={"inputs": prompt}, timeout=120)