Yhhxhfh commited on
Commit
a201f43
1 Parent(s): 3bac2a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -64,9 +64,15 @@ async def train_and_save_model():
64
  user_questions = r.lrange('user_questions', 0, -1)
65
  for question in user_questions:
66
  question = question.decode('utf-8')
67
- documents.append((nltk.word_tokenize(question), "unknown"))
68
- if "unknown" not in classes:
69
- classes.append("unknown")
 
 
 
 
 
 
70
  r.set('user_questions_loaded', 1)
71
 
72
  for intent in intents['intents']:
 
64
  user_questions = r.lrange('user_questions', 0, -1)
65
  for question in user_questions:
66
  question = question.decode('utf-8')
67
+ try:
68
+ existing_tag = r.get(f"tag:{question}").decode('utf-8')
69
+ documents.append((nltk.word_tokenize(question), existing_tag))
70
+ if existing_tag not in classes:
71
+ classes.append(existing_tag)
72
+ except AttributeError:
73
+ documents.append((nltk.word_tokenize(question), "unknown"))
74
+ if "unknown" not in classes:
75
+ classes.append("unknown")
76
  r.set('user_questions_loaded', 1)
77
 
78
  for intent in intents['intents']: