fix unneeded import

ps4_models/classifiers.py

@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-from ps4_data.utils import SS_CLASSES
 from mega.fairseq.modules.mega_layer import MegaEncoderLayer
 
 
@@ -53,7 +52,7 @@ class PS4_Mega(nn.Module):
 
         self.args = ArgHolder(emb_dim=self.input_size, dropout=dropout, hdim=h_dim)
 
-        self.nb_tags = SS_CLASSES
+        self.nb_tags = 8
 
         self.model_parallel = model_parallel