diff --git a/src/huggingface_inference_toolkit/serialization/base.py b/src/huggingface_inference_toolkit/serialization/base.py index 52eb0385..731139c4 100644 --- a/src/huggingface_inference_toolkit/serialization/base.py +++ b/src/huggingface_inference_toolkit/serialization/base.py @@ -10,6 +10,7 @@ # image types "image/png": Imager, "image/jpeg": Imager, + "image/jpg": Imager, "image/tiff": Imager, "image/bmp": Imager, "image/gif": Imager, @@ -17,8 +18,11 @@ "image/x-image": Imager, # audio types "audio/x-flac": Audioer, + "audio/flac": Audioer, "audio/mpeg": Audioer, "audio/wave": Audioer, + "audio/wav": Audioer, + "audio/x-wav": Audioer, "audio/ogg": Audioer, "audio/x-audio": Audioer, }