Update app.py

app.py

@@ -31,6 +31,20 @@ model.load("MentalHealthChatBotmodel.tflearn")
 # print('model loaded successfully')
 
 
+def bag_of_words(s, words):
+    bag = [0 for _ in range(len(words))]
+
+    s_words = nltk.word_tokenize(s)
+    s_words = [stemmer.stem(word.lower()) for word in s_words]
+
+    for se in s_words:
+        for i, w in enumerate(words):
+            if w == se:
+                bag[i] = 1
+            
+    return np.array(bag)
+
+
 def chat(message, history):
     history = history or []
     message = message.lower()