Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

dda2ec1

1 Parent(s): f3435ae

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -455,3 +455,4 @@ class Summarization:
     def evaluate(
             self,
             test_df: pd.DataFrame,

     def evaluate(
             self,
             test_df: pd.DataFrame,
+            metrics: str = "rouge"