Update modeling_provence.py

modeling_provence.py

@@ -99,8 +99,8 @@ class Provence(DebertaV2PreTrainedModel):
 
     def process(
         self,
+        question: Union[List[str], str],
         context: Union[List[List[str]], List[str], str],
-        query: Union[List[str], str],
         title: Optional[Union[List[List[str]], List[str], str]] = "first_sentence",
         batch_size=32,
         threshold=0.1,
@@ -111,10 +111,10 @@ class Provence(DebertaV2PreTrainedModel):
     ):
 
         # convert input format into queries of type List[str] and contexts/titles of type List[List[str]]
-        if type(query) == str:
-            queries = [query]
+        if type(question) == str:
+            queries = [question]
         else: # list of strs
-            queries = query
+            queries = question
         if type(context) == str:
             contexts = [[context]]
         elif type(context) == list and type(context[0]) == str: