fix line 99

infer_concat.py

@@ -96,8 +96,7 @@ def infer_2_hier(model, data_loader, device, tokenizer):
                 summaries.append(summary)
         summaries = torch.cat(summaries, dim = 1)
 	
-        all_summaries.append(tokenizer.decode(summaries, skip_special_tokens=True))
-
+        all_summaries.append(tokenizer.decode(summaries.squeeze(), skip_special_tokens=True))
     
     end = time.time()
     print(f"Time: {end-start}")