Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 23, 2021

Commit

5ffa1ac

•

1 Parent(s): 5787075

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -457,3 +457,4 @@ class Summarization:
             test_df: pd.DataFrame,
             metrics: str = "rouge"
     ):

             test_df: pd.DataFrame,
             metrics: str = "rouge"
     ):
+        metric = load_metric(metrics)