From 6f0c88b1ac31b424de5af46cade5b6f1fb816879 Mon Sep 17 00:00:00 2001 From: bobloy Date: Thu, 1 Jul 2021 17:19:29 -0400 Subject: [PATCH] change default models, fix requirements --- chatter/chat.py | 9 +++++---- chatter/info.json | 1 + 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/chatter/chat.py b/chatter/chat.py index 2deb082..e3f14cc 100644 --- a/chatter/chat.py +++ b/chatter/chat.py @@ -76,7 +76,7 @@ class Chatter(Cog): # TODO: Move training_model and similarity_algo to config # TODO: Add an option to see current settings - self.tagger_language = ENG_TRF + self.tagger_language = ENG_SM self.similarity_algo = SpacySimilarity self.similarity_threshold = 0.90 self.chatbot = self._create_chatbot() @@ -85,7 +85,7 @@ class Chatter(Cog): # self.trainer = ListTrainer(self.chatbot) self.config.register_global(**default_global) - self.config.register_guild(**default_guild) + self.config.register_guild(**self.default_guild) self.loop = asyncio.get_event_loop() @@ -371,11 +371,12 @@ class Chatter(Cog): 0: Small 1: Medium 2: Large (Requires additional setup) + 3. Accurate (Requires additional setup) """ - models = [ENG_SM, ENG_MD, ENG_LG] + models = [ENG_SM, ENG_MD, ENG_LG, ENG_TRF] - if model_number < 0 or model_number > 2: + if model_number < 0 or model_number > 3: await ctx.send_help() return diff --git a/chatter/info.json b/chatter/info.json index a9bb96b..0004709 100644 --- a/chatter/info.json +++ b/chatter/info.json @@ -8,6 +8,7 @@ "install_msg": "Thank you for installing Chatter! Please make sure you check the install instructions at https://github.com/bobloy/Fox-V3/blob/master/chatter/README.md\nAfter that, get started ith `[p]load chatter` and `[p]help Chatter`", "requirements": [ "git+git://github.com/bobloy/ChatterBot@fox#egg=ChatterBot", + "en_core_web_sm", "kaggle" ], "short": "Local Chatbot run on machine learning",