Update app.py

app.py

@@ -8,7 +8,10 @@ from deep_translator import GoogleTranslator
 
 app = Flask(__name__)
 
-API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
+API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
+
+API_TOKEN = os.getenv("HF_READ_TOKEN")
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 50000  # タイムアウトを300秒に設定
 
 # Function to query the API and return the generated image
@@ -42,7 +45,7 @@ def query(prompt, negative_prompt="", steps=35, cfg_scale=7, sampler="DPM++ 2M K
 
     for attempt in range(3):  # 最大3回の再試行
         try:
-            response = requests.post(API_URL, json=payload, timeout=timeout)
+            response = requests.post(API_URL, json=payload, headers=headers, timeout=timeout)
             if response.status_code != 200:
                 return None, f"Error: Failed to get image. Status code: {response.status_code}, Details: {response.text}"