Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
daf0155
1
Parent(s):
3e38b94
Update model.py
Browse files- src/models/model.py +3 -1
src/models/model.py
CHANGED
@@ -473,7 +473,9 @@ class Summarization:
|
|
473 |
metric = load_metric(metrics)
|
474 |
input_text = test_df['input_text']
|
475 |
references = test_df['output_text']
|
476 |
-
|
|
|
|
|
477 |
|
478 |
results = metric.compute(predictions=predictions, references=references)
|
479 |
|
|
|
473 |
metric = load_metric(metrics)
|
474 |
input_text = test_df['input_text']
|
475 |
references = test_df['output_text']
|
476 |
+
references = references.to_list()
|
477 |
+
|
478 |
+
predictions = [self.predict(x) for x in tqdm(input_text)]
|
479 |
|
480 |
results = metric.compute(predictions=predictions, references=references)
|
481 |
|