From c8733390e23e22c0a7ca97cd95af543e84b38519 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Thu, 3 Jul 2025 16:54:40 +0200
Subject: [PATCH] Somehow handle rate limit

---
 facefusion_api/core.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/facefusion_api/core.py b/facefusion_api/core.py
index be22699..9b0c673 100644
--- a/facefusion_api/core.py
+++ b/facefusion_api/core.py
@@ -79,9 +79,12 @@ class SwapFaceImage:
 		with HttpClient(transport = transport) as http_client:
 			response = http_client.post(url, headers = headers, files = files, data = data)
 
-		output_buffer = BytesIO(response.content)
-		output_tensor = bytesio_to_image_tensor(output_buffer)
-		return output_tensor
+			if response.status_code == 200:
+				output_buffer = BytesIO(response.content)
+				output_tensor = bytesio_to_image_tensor(output_buffer)
+				return output_tensor
+
+		return target_tensor
 
 
 class SwapFaceVideo: