Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
502f04f
1
Parent(s):
7c59938
Update model.py
Browse files- src/models/model.py +1 -5
src/models/model.py
CHANGED
@@ -488,11 +488,7 @@ class Summarization:
|
|
488 |
)
|
489 |
return preds
|
490 |
|
491 |
-
def evaluate(
|
492 |
-
self,
|
493 |
-
test_df: pd.DataFrame,
|
494 |
-
metrics: str = "rouge"
|
495 |
-
):
|
496 |
metric = load_metric(metrics)
|
497 |
input_text = test_df['input_text']
|
498 |
references = test_df['output_text']
|
|
|
488 |
)
|
489 |
return preds
|
490 |
|
491 |
+
def evaluate(self, test_df: pd.DataFrame, metrics: str = "rouge"):
|
|
|
|
|
|
|
|
|
492 |
metric = load_metric(metrics)
|
493 |
input_text = test_df['input_text']
|
494 |
references = test_df['output_text']
|