Spaces:

gagan3012
/

T5-Summarization

Runtime error

Gagan Bhatia commited on Jul 14, 2021

Commit

8e6d41d

1 Parent(s): 5c300b0

Update model.py

Files changed (1) hide show

src/models/model.py CHANGED Viewed

@@ -30,3 +30,6 @@ class DataModule(Dataset):
             data: pd.DataFrame,
             tokenizer: T5Tokenizer,
             source_max_token_len: int = 512,

             data: pd.DataFrame,
             tokenizer: T5Tokenizer,
             source_max_token_len: int = 512,
+            target_max_token_len: int = 512,
+    ):
+        """