diff --git a/src/ProcessAnnotator.py b/src/ProcessAnnotator.py
index e01b9a3..71897bb 100644
--- a/src/ProcessAnnotator.py
+++ b/src/ProcessAnnotator.py
@@ -94,7 +94,7 @@ def process(self, flow):
     def __predict_label(self, flow):
         df = pd.DataFrame(columns=flow.parameters.keys(), data=[flow.parameters.values()])
         df.replace('', np.nan, inplace=True)
-        # y_pred = self.ml_model.predict_proba(df, verbose=0)
+        y_pred = self.ml_model.predict_proba(df, verbose=0)
         # y_pred_classes = np.argmax(y_pred, axis=0)
         # flow.add_parameter(Config.Texts.Prediction, self.label_index[str(y_pred_classes)])
         # flow.add_parameter(Config.Texts.prediction_confidence, str(y_pred[y_pred_classes]))