Update app.py
Browse files
app.py
CHANGED
@@ -126,8 +126,8 @@ async def train_and_save_model():
|
|
126 |
await asyncio.sleep(60)
|
127 |
continue
|
128 |
|
129 |
-
train_x = np.array([row[0] for row in training]
|
130 |
-
train_y = np.array([row[1] for row in training]
|
131 |
|
132 |
print("Loading or creating model...")
|
133 |
if os.path.exists('chatbot_model.h5'):
|
@@ -189,7 +189,7 @@ async def chat(message: ChatMessage):
|
|
189 |
if w == s:
|
190 |
bag[i] = 1
|
191 |
|
192 |
-
p = model.predict(np.array([bag]
|
193 |
ERROR_THRESHOLD = 0.25
|
194 |
results = [[i, p] for i, p in enumerate(p) if p > ERROR_THRESHOLD]
|
195 |
results.sort(key=lambda x: x[1], reverse=True)
|
|
|
126 |
await asyncio.sleep(60)
|
127 |
continue
|
128 |
|
129 |
+
train_x = np.array([row[0] for row in training])
|
130 |
+
train_y = np.array([row[1] for row in training])
|
131 |
|
132 |
print("Loading or creating model...")
|
133 |
if os.path.exists('chatbot_model.h5'):
|
|
|
189 |
if w == s:
|
190 |
bag[i] = 1
|
191 |
|
192 |
+
p = model.predict(np.array([bag]))[0]
|
193 |
ERROR_THRESHOLD = 0.25
|
194 |
results = [[i, p] for i, p in enumerate(p) if p > ERROR_THRESHOLD]
|
195 |
results.sort(key=lambda x: x[1], reverse=True)
|