Minor correction

This commit is contained in:
netkas 2020-12-27 23:39:53 -05:00
parent c39f934912
commit 7c40272f24
1 changed files with 1 additions and 1 deletions

View File

@ -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):