Upload app.py with huggingface_hub

app.py

@@ -137,7 +137,7 @@ class WordConstraintLogitsProcessor(LogitsProcessor):
         self.constraints = constraints
         self.prompt_len = prompt_len
 
-    def __call__(self, input_ids, scores, **kwargs):
+    def __call__(self, input_ids, scores, cur_step=None):
         if scores.dim() != 3:
             return scores
         committed_len = input_ids.shape[1] - self.prompt_len