Yhhxhfh commited on
Commit
8910a27
1 Parent(s): bd9d99c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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], dtype=np.float32)
130
- train_y = np.array([row[1] for row in training], dtype=np.float32)
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], dtype=np.float32))[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)
 
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)