fix: update type for image translation.

Author: Khurdhula-Harshavardhan

jigsawstack/translate.py

@@ -146,17 +146,17 @@ async def text(self, params: TranslateParams) -> TranslateResponse:
         return resp
 
     @overload
-    async def image(self, params: TranslateImageParams) -> TranslateImageResponse: ...
+    async def image(self, params: TranslateImageParams) -> Union[TranslateImageResponse, bytes]: ...
     @overload
     async def image(
         self, blob: bytes, options: TranslateImageParams = None
-    ) -> TranslateImageParams: ...
+    ) -> Union[TranslateImageResponse, bytes]: ...
 
     async def image(
         self,
         blob: Union[TranslateImageParams, bytes],
         options: TranslateImageParams = None,
-    ) -> TranslateImageResponse:
+    ) -> Union[TranslateImageResponse, bytes]:
         if isinstance(blob, dict):
             resp = await AsyncRequest(
                 config=self.config,