Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
bf6e21d
1
Parent(s): 5941304
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
|
@@ -460,3 +460,4 @@ class Summarization:
|
|
| 460 |
metric = load_metric(metrics)
|
| 461 |
input_text = test_df['input_text']
|
| 462 |
references = test_df['output_text']
|
|
|
|
|
|
| 460 |
metric = load_metric(metrics)
|
| 461 |
input_text = test_df['input_text']
|
| 462 |
references = test_df['output_text']
|
| 463 |
+
predictions = [self.predict(x) for x in input_text]
|