Update app.py

app.py

@@ -13,8 +13,14 @@ headers = {
     "Authorization": f"Bearer {HF_API_KEY}"
 }
 
-def call_hf_api(api_url, payload):
-    response = requests.post(api_url, headers=headers, json=payload)
+def call_hf_api(api_url, image_bytes, parameters=None):
+    # 画像データを multipart/form-data として送信
+    files = {
+        "file": ("image.png", image_bytes, "image/png"),
+    }
+    data = {"parameters": parameters} if parameters else {}
+
+    response = requests.post(api_url, headers=headers, files=files, data=data)
     if response.status_code != 200:
         raise Exception(f"Request failed with status {response.status_code}, {response.text}")
     return response.json()
@@ -25,19 +31,15 @@ def process_image_with_api(image_path):
         image_bytes = f.read()
     
     # Prior Reduxモデルで事前処理
-    prior_payload = {"inputs": image_bytes}
-    prior_response = call_hf_api(API_URL_PRIOR, prior_payload)
+    prior_response = call_hf_api(API_URL_PRIOR, image_bytes)
     
     # FLUXモデルで画像生成
     flux_payload = {
-        "inputs": prior_response,  # Prior Reduxの出力をFLUXモデルに渡す
-        "parameters": {
-            "guidance_scale": 2.5,
-            "num_inference_steps": 50,
-            "seed": 0,  # 再現性のためのシード値
-        }
+        "guidance_scale": 2.5,
+        "num_inference_steps": 50,
+        "seed": 0,  # 再現性のためのシード値
     }
-    flux_response = call_hf_api(API_URL_FLUX, flux_payload)
+    flux_response = call_hf_api(API_URL_FLUX, image_bytes, parameters=flux_payload)
     
     # 生成された画像を取得
     generated_image_url = flux_response.get("generated_image_url")