Update model.py

src/models/model.py

@@ -490,8 +490,8 @@ class Summarization:
 
     def evaluate(self, test_df: pd.DataFrame, metrics: str = "rouge"):
         metric = load_metric(metrics)
-        input_text = test_df['input_text']
-        references = test_df['output_text']
+        input_text = test_df["input_text"]
+        references = test_df["output_text"]
         references = references.to_list()
 
         predictions = [self.predict(x) for x in tqdm(input_text)]