Spaces:

gagan3012
/

summarization

Runtime error

gagan3012 commited on Jul 26, 2021

Commit

a49bf5d

1 Parent(s): 5d9f40a

fixes

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -478,7 +478,7 @@ class Summarization:
         references = test_df['output_text']
         references = references.to_list()
-        predictions = [self.predict(x) for x in input_text]
         results = metric.compute(predictions=predictions, references=references)

         references = test_df['output_text']
         references = references.to_list()
+        predictions = [self.predict(x) for x in tqdm(input_text)]
         results = metric.compute(predictions=predictions, references=references)