Arafath10 commited on
Commit
efdc6c0
·
1 Parent(s): 59c62ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -2,12 +2,18 @@ import gradio as gr
2
  from chatterbot import ChatBot
3
  from chatterbot.trainers import ListTrainer
4
  #from chatterbot.trainers import ChatterBotCorpusTrainer
 
5
 
6
  chatbot = ChatBot("Chatpot")
7
 
8
  trainer = ListTrainer(chatbot)
9
 
10
- trainer.train("dialogs.txt")
 
 
 
 
 
11
 
12
 
13
 
 
2
  from chatterbot import ChatBot
3
  from chatterbot.trainers import ListTrainer
4
  #from chatterbot.trainers import ChatterBotCorpusTrainer
5
+ from cleaner import clean_corpus
6
 
7
  chatbot = ChatBot("Chatpot")
8
 
9
  trainer = ListTrainer(chatbot)
10
 
11
+ CORPUS_FILE = "chat.txt"
12
+
13
+ trainer = ListTrainer(chatbot)
14
+ cleaned_corpus = clean_corpus(CORPUS_FILE)
15
+ trainer.train(cleaned_corpus)
16
+ print(cleaned_corpus)
17
 
18
 
19