added extra_headers to query_in_df

MinuraPunchihewa · Jul 17, 2024 · c508c11 · c508c11
1 parent 57485e2
commit c508c11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hugging_py_face/nlp.py b/hugging_py_face/nlp.py
@@ -59,7 +59,7 @@ def _query(self, inputs: Union[Text, List, Dict], parameters: Optional[Dict] = N
         raise HTTPServiceUnavailableException("The HTTP service is unavailable.")
 
     def _query_in_df(self, df: DataFrame, column: Text, parameters: Optional[Dict] = None, options: Optional[Dict] = None, model: Optional[Text] = None, task: Optional[Text] = None, extra_headers: Optional[Dict] = None) -> Union[Dict, List]:
-        return self._query(df[column].tolist(), parameters, options, model, task)
+        return self._query(df[column].tolist(), parameters, options, model, task, extra_headers)
 
     def fill_mask(self, text: Union[Text, List], options: Optional[Dict] = None, model: Optional[Text] = None) -> List:
         """