Update model.py

src/models/model.py

@@ -65,3 +65,7 @@ class DataModule(Dataset):
             padding="max_length",
             truncation=True,
             return_attention_mask=True,
+            add_special_tokens=True,
+            return_tensors="pt",
+        )
+