Added some more models

2022-04-03 19:41:03 +03:00
parent 9ba51c0fd1
commit c0d2b1e35b
2 changed files with 10 additions and 3 deletions
--- a/create_models.py
+++ b/create_models.py
@@ -9,7 +9,10 @@ api_id = os.getenv('API_ID')
 api_hash = os.getenv('API_HASH')
 chats = [
  'tg1337const',
-  'cyberboec'
+  'cyberboec',
+  'topor',
+  'workBrothers',
+  'anekdot_bb'
 ]

 def cleanup(msg):
@@ -35,7 +38,7 @@ with TelegramClient('bot', api_id, api_hash) as client:
        data = data + cleanup(message.text)

    text_model = markovify.Text(data)
-    # text_model.compile(inplace = True)
+    text_model.compile(inplace = True)
    model_json = text_model.to_json()
    f = open(f"data/{chat}.json", "w")
    f.write(model_json)