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