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 |
|