Somehow handle rate limit

This commit is contained in:
henryruhs
2025-07-03 16:54:40 +02:00
parent 620910097b
commit c8733390e2
+6 -3
View File
@@ -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: