Update app.py

app.py

@@ -73,13 +73,14 @@ def remove_watermark(img_bytes):
         API_HOST,
         files=files,
     )
+    print(resp)
     
     if resp.status_code != 200:
         print(f"Error: {resp.status_code} - {resp.text}")
         return None
 
-    base64_image = resp.json().get("edited_image").get("image")
-    return base64.b64decode(base64_image)
+    result_img = Image.open(io.BytesIO(resp.content)).convert("RGB")
+    return result_img
 
 def process(input_image):
     """
@@ -98,7 +99,7 @@ def process(input_image):
         # return None  # or return input_image as fallback
 
     # Convert bytes back to PIL image
-    result_img = Image.open(io.BytesIO(result_bytes)).convert("RGB")
+    # result_img = Image.open(io.BytesIO(result_bytes)).convert("RGB")
     return [input_image, result_img]
     
 with gr.Blocks(css=css) as demo: