Hugging Face's logo

Spaces:
gagan3012
/
T5-Summarization
Runtime error

App Files Community
Gagan Bhatia commited on
Commit
5c300b0
•
1 Parent(s): 336121a

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -27,3 +27,6 @@ class DataModule(Dataset):
27
 
28
  def __init__(
29
  self,
 
 
 
 
27
 
28
  def __init__(
29
  self,
30
+ data: pd.DataFrame,
31
+ tokenizer: T5Tokenizer,
32
+ source_max_token_len: int = 512,