Update app.py

app.py

@@ -72,17 +72,17 @@ def process_image_and_text(subject_image, adapter_dict, prompt, _unused1, _unuse
     resp = requests.post(BASETEN_MODEL_URL, headers=headers, json=payload, timeout=180)
     resp.raise_for_status()
     data = resp.json()
-    print (data)
     
-    # If Baseten returns [cond1.condition, raw_img] format
-    if isinstance(data, list) and len(data) == 2:
-        # If the second item is base64-encoded image string
+    # Extract base64 image from 'blended' key
+    if "blended" in data:
         try:
-            raw_img = Image.open(BytesIO(base64.b64decode(data[1])))
+            blended_bytes = base64.b64decode(data["blended"])
+            raw_img = Image.open(BytesIO(blended_bytes)).convert("RGB")
         except Exception:
-            raise gr.Error("Failed to decode image from Baseten response.")
+            raise gr.Error("Failed to decode 'blended' image from Baseten response.")
     else:
-        raise gr.Error("Unexpected response format from Baseten.")
+        raise gr.Error("Baseten response missing 'blended' image.")
+
 
 
 # ────────── Header HTML ──────────