Gagan Bhatia commited on
Commit
da6f7d4
1 Parent(s): a51b600

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +5 -5
src/models/model.py CHANGED
@@ -31,11 +31,11 @@ class DataModule(Dataset):
31
  """
32
 
33
  def __init__(
34
- self,
35
- data: pd.DataFrame,
36
- tokenizer: T5Tokenizer,
37
- source_max_token_len: int = 512,
38
- target_max_token_len: int = 512,
39
  ):
40
  """
41
  :param data:
 
31
  """
32
 
33
  def __init__(
34
+ self,
35
+ data: pd.DataFrame,
36
+ tokenizer: T5Tokenizer,
37
+ source_max_token_len: int = 512,
38
+ target_max_token_len: int = 512,
39
  ):
40
  """
41
  :param data: