code: def predict(self, text):
          
          pred = self.predict_proba(text)
          tags = self._get_tags(pred)
  
          return tags
docs: Predict using the model.
  
          Args:
              text: string, the input text.
  
          Returns:
              tags: list, shape = (num_words,)
              Returns predicted values.