Allow specifying data with the Inference API wrapper

src/huggingface_hub/inference_api.py

@@ -122,20 +122,23 @@ def __repr__(self):
 
     def __call__(
         self,
-        inputs: Union[str, Dict, List[str], List[List[str]]],
+        inputs: Optional[Union[str, Dict, List[str], List[List[str]]]] = None,
         params: Optional[Dict] = None,
+        data: Optional[bytes] = None,
     ):
         payload = {
-            "inputs": inputs,
             "options": self.options,
         }
 
+        if inputs:
+            payload["inputs"] = inputs
+
         if params:
             payload["parameters"] = params
 
         # TODO: Decide if we should raise an error instead of
         # returning the json.
         response = requests.post(
-            self.api_url, headers=self.headers, json=payload
+            self.api_url, headers=self.headers, json=payload, data=data
         ).json()
         return response

tests/test_inference_api.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-
+import os
 import unittest
 
 from huggingface_hub.inference_api import InferenceApi
@@ -21,6 +21,13 @@
 
 
 class InferenceApiTest(unittest.TestCase):
+    def read(self, filename: str) -> bytes:
+        dirname = os.path.dirname(os.path.abspath(__file__))
+        filename = os.path.join(dirname, "samples", filename)
+        with open(filename, "rb") as f:
+            bpayload = f.read()
+        return bpayload
+
     @with_production_testing
     def test_simple_inference(self):
         api = InferenceApi("bert-base-uncased")
@@ -55,6 +62,25 @@ def test_inference_with_dict_inputs(self):
         self.assertTrue("score" in result)
         self.assertTrue("answer" in result)
 
+    @with_production_testing
+    def test_inference_with_audio(self):
+        api = InferenceApi("facebook/wav2vec2-large-960h-lv60-self")
+        data = self.read("sample1.flac")
+        result = api(data=data)
+        self.assertIsInstance(result, dict)
+        self.assertTrue("text" in result)
+
+    @with_production_testing
+    def test_inference_with_image(self):
+        api = InferenceApi("google/vit-base-patch16-224")
+        data = self.read("plane.jpg")
+        result = api(data=data)
+        self.assertIsInstance(result, list)
+        for classification in result:
+            self.assertIsInstance(classification, dict)
+            self.assertTrue("score" in classification)
+            self.assertTrue("label" in classification)
+
     @with_production_testing
     def test_inference_overriding_task(self):
         api = InferenceApi(