fix

.gitignore

@@ -8,3 +8,5 @@ binoculars/__pycache__/detector.cpython-310.pyc
 binoculars/__pycache__/__init__.cpython-310.pyc
 __pycache__/text_analysis.cpython-310.pyc
 __pycache__/binoculars_utils.cpython-310.pyc
+NN_classifier/__pycache__/neural_net_t.cpython-310.pyc
+demo/__pycache__/__init__.cpython-310.pyc

model_utils.py

@@ -67,7 +67,7 @@ def load_ternary_model(model_dir='models/neural_network'):
     input_size = scaler.n_features_in_
     num_classes = len(label_encoder.classes_)
     
-    model = Neural_Network(input_size, hidden_layers=[256, 192, 128, 64], num_classes=num_classes, dropout_rate=0.3).to(DEVICE)
+    model = Neural_Network(input_size, hidden_layers=[128, 96, 64, 32], num_classes=num_classes, dropout_rate=0.1).to(DEVICE)
     model.load_state_dict(torch.load(model_path, map_location=DEVICE))
     model.eval()