maybe fix pipeline

pipeline.py

@@ -20,8 +20,8 @@ class PreTrainedPipeline():
               A :obj:`list` | `dict`: will be serialized and returned
         """
         # compute the embedding of the input
-        query = data["inputs"]
-        inputs = self.tokenizer(query, padding=True, return_tensors="pt").to(
+        # query = data["inputs"]
+        inputs = self.tokenizer(data, padding=True, return_tensors="pt").to(
             self.device
         )
         with torch.no_grad():