Update model.py

model.py

@@ -9,7 +9,7 @@ class ClassificationModel(nn.Module):
             nn.ReLU(),
             nn.Dropout(0.3),
             nn.Linear(256, 8),
-            nn.Softmax(dim=1)
+            nn.LogSoftmax(dim=1)
         )
 
     def forward(self, input_ids, attention_mask):