diff --git a/services/spam_detection/coffeehouse_spamdetection/main.py b/services/spam_detection/coffeehouse_spamdetection/main.py index 24c18f86..3505a7af 100644 --- a/services/spam_detection/coffeehouse_spamdetection/main.py +++ b/services/spam_detection/coffeehouse_spamdetection/main.py @@ -13,7 +13,7 @@ class SpamDetection(object): """ self.dltc = DLTC() self.rf = ResourceFetch() - self.model_directory = os.path.join(self.rf.fetch("Intellivoid", "CoffeeHouseData-Spam"), 'spam_ham_build') + self.model_directory = os.path.join(self.rf.fetch("Intellivoid", "CoffeeHouseData-Spam"), 'chatrooms_build') self.dltc.load_model_cluster(self.model_directory) def predict(self, text_input):