Spaces:

kleinay
/

qanom-seq2seq-demo

Runtime error

kleinay commited on May 10, 2022

Commit

1ae67ef

1 Parent(s): 1c27898

Update qasrl_model_pipeline.py

Files changed (1) hide show

qasrl_model_pipeline.py CHANGED Viewed

@@ -147,7 +147,7 @@ class QASRL_Pipeline(Text2TextGenerationPipeline):
     def postprocess(self, model_outputs):
         output_seq = self.tokenizer.decode(
-            model_outputs["output_ids"][0],
             skip_special_tokens=False,
             clean_up_tokenization_spaces=False,
         )

     def postprocess(self, model_outputs):
         output_seq = self.tokenizer.decode(
+            model_outputs["output_ids"].squeeze(),
             skip_special_tokens=False,
             clean_up_tokenization_spaces=False,
         )