Prob fix

.gitignore

@@ -1 +1,2 @@
-.idea/
+.idea/
+**/__pycache__/**

models/bayes.py

@@ -2,7 +2,6 @@ from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.preprocessing import MultiLabelBinarizer
 import typing
 import joblib
-from utils.preprocessing import clean
 
 
 class NaiveBayesMultiClass(object):
@@ -47,7 +46,7 @@ class NaiveBayesMultiClass(object):
         x = self.vectorizer.transform(X)
         by_class = [self.classifiers[i].predict_proba(x) for i in range(self.n_classes)]
 
-        return [[by_class[j][i] for j in range(self.n_classes)] for i in range(len(X))]  # type: ignore
+        return [[by_class[j][i][1] for j in range(self.n_classes)] for i in range(len(X))]  # type: ignore
 
 
 def get_tags_bayes(model: NaiveBayesMultiClass, text: str, threshold : None | float =None):