Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
15505de
1
Parent(s): fae8953
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -44,3 +44,6 @@ class DataModule(Dataset):
|
|
| 44 |
self.tokenizer = tokenizer
|
| 45 |
|
| 46 |
def __len__(self):
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
self.tokenizer = tokenizer
|
| 45 |
|
| 46 |
def __len__(self):
|
| 47 |
+
return len(self.data)
|
| 48 |
+
|
| 49 |
+
def __getitem__(self, index: int):
|