Yhhxhfh commited on
Commit
14fcfac
1 Parent(s): a2a58e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -133,13 +133,14 @@ async def train_and_save_model():
133
  if os.path.exists('chatbot_model.h5'):
134
  model = load_model('chatbot_model.h5')
135
  else:
136
- model = Sequential()
137
- model.add(Input(shape=(len(train_x[0]),)))
138
- model.add(Dense(128, activation='relu'))
139
- model.add(Dropout(0.5))
140
- model.add(Dense(64, activation='relu'))
141
- model.add(Dropout(0.5))
142
- model.add(Dense(len(classes), activation='softmax'))
 
143
 
144
  sgd = SGD(learning_rate=0.01, momentum=0.9, nesterov=True)
145
  model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy'])
 
133
  if os.path.exists('chatbot_model.h5'):
134
  model = load_model('chatbot_model.h5')
135
  else:
136
+ input_layer = Input(shape=(len(train_x[0]),))
137
+ layer1 = Dense(128, activation='relu')(input_layer)
138
+ layer2 = Dropout(0.5)(layer1)
139
+ layer3 = Dense(64, activation='relu')(layer2)
140
+ layer4 = Dropout(0.5)(layer3)
141
+ output_layer = Dense(len(classes), activation='softmax')(layer4)
142
+
143
+ model = Sequential(layers=[input_layer, layer1, layer2, layer3, layer4, output_layer])
144
 
145
  sgd = SGD(learning_rate=0.01, momentum=0.9, nesterov=True)
146
  model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy'])