fix double sigmoid issue in mlp

Author: aditya0by0

chebai/models/ffn.py

@@ -26,7 +26,6 @@ def __init__(
             current_layer_input_size = hidden_dim
 
         layers.append(torch.nn.Linear(current_layer_input_size, self.out_dim))
-        layers.append(nn.Sigmoid())
         self.model = nn.Sequential(*layers)
 
     def _get_prediction_and_labels(self, data, labels, model_output):