Skip to content

Commit c873339

Browse files
committed
Somehow handle rate limit
1 parent 6209100 commit c873339

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

facefusion_api/core.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,9 +79,12 @@ def swap_face(source_tensor : Tensor, target_tensor : Tensor, api_token : str, f
7979
with HttpClient(transport = transport) as http_client:
8080
response = http_client.post(url, headers = headers, files = files, data = data)
8181

82-
output_buffer = BytesIO(response.content)
83-
output_tensor = bytesio_to_image_tensor(output_buffer)
84-
return output_tensor
82+
if response.status_code == 200:
83+
output_buffer = BytesIO(response.content)
84+
output_tensor = bytesio_to_image_tensor(output_buffer)
85+
return output_tensor
86+
87+
return target_tensor
8588

8689

8790
class SwapFaceVideo:

0 commit comments

Comments
 (0)