update

app.py

@@ -8,20 +8,21 @@ import os
 
 def call_predict_api(person, cloth):
     # Convert PIL images to base64 string
-    def pil_to_b64(img):
-        buffer = BytesIO()
-        img.save(buffer, format='PNG')
-        return base64.b64encode(buffer.getvalue()).decode('utf-8')
+    src_buffer = BytesIO()
+    person.save(src_buffer, format='PNG')
+    src_buffer.seek(0)
     
-    person = pil_to_b64(person)
-    cloth = pil_to_b64(cloth)
+    ref_buffer = BytesIO()
+    cloth.save(ref_buffer, format='PNG')
+    ref_buffer.seek(0)
     
-    payload = {
-        "src_image_base64": person,
-        "ref_image_base64": cloth
+    # Prepare files for upload
+    files = {
+        "src_image": ("src_image.png", src_buffer, "image/png"),
+        "ref_image": ("ref_image.png", ref_buffer, "image/png")
     }
     headers = {"X-API-Key": os.environ["api_key"]}
-    response = requests.post(os.environ["endpoint"], json=payload, headers=headers)
+    response = requests.post(os.environ["endpoint"], files=files, headers=headers)
     if response.status_code != 200:
         raise Exception(f"API Error: {response.text}")
     result = response.json()