Update main.py

main.py

@@ -13,7 +13,7 @@ from dataset import SpeechesClassificationDataset
 
 from constants import seed, batch_size, block_size, learning_rate, n_embd, n_head, n_layer, n_input, n_output, n_hidden, epochs_CLS
 
-from transformer import Classifier, Decoder, DecoderEC
+from transformer import Classifier
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")