Update app.py

app.py

@@ -163,12 +163,14 @@ async def train_and_save_model():
             model = load_model(temp_file_name)
             os.remove(temp_file_name)
         else:
-            model = Sequential()
-            model.add(Dense(128, input_shape=(len(train_x[0]),), activation='relu'))
-            model.add(Dropout(0.5))
-            model.add(Dense(64, activation='relu'))
-            model.add(Dropout(0.5))
-            model.add(Dense(len(train_y[0]), activation='softmax'))
+            input_layer = Input(shape=(len(train_x[0]),)) 
+            layer1 = Dense(128, activation='relu')(input_layer)
+            layer2 = Dropout(0.5)(layer1)
+            layer3 = Dense(64, activation='relu')(layer2)
+            layer4 = Dropout(0.5)(layer3)
+            output_layer = Dense(len(classes), activation='softmax')(layer4)
+
+            model = Sequential(layers=[input_layer, layer1, layer2, layer3, layer4, output_layer])
 
             sgd = SGD(learning_rate=0.01, momentum=0.9, nesterov=True)
             model.compile(loss='categorical_crossentropy', optimizer=sgd, metrics=['accuracy'])