Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -140,7 +140,7 @@ class CustomDataset(torch.utils.data.Dataset):
|
|
140 |
text,
|
141 |
max_length=self.max_length,
|
142 |
padding='max_length',
|
143 |
-
|
144 |
return_attention_mask=True,
|
145 |
return_tensors='pt',
|
146 |
)
|
|
|
140 |
text,
|
141 |
max_length=self.max_length,
|
142 |
padding='max_length',
|
143 |
+
truncation=True,
|
144 |
return_attention_mask=True,
|
145 |
return_tensors='pt',
|
146 |
)
|