fixes

src/models/model.py

@@ -15,7 +15,6 @@ from pytorch_lightning import LightningModule
 from datasets import load_metric
 from tqdm.auto import tqdm
 
-
 # from dagshub.pytorch_lightning import DAGsHubLogger
 
 
@@ -477,9 +476,10 @@ class Summarization:
         metric = load_metric(metrics)
         input_text = test_df['input_text'][:5]
         references = test_df['output_text'][:5]
+        references = references.to_list()
 
         predictions = [self.predict(x) for x in input_text]
-        print(type(predictions),type(references))
+        print(type(predictions), type(references))
 
         results = metric.compute(predictions=predictions, references=references)
         '''