cheenchan commited on
Commit
f977328
·
verified ·
1 Parent(s): 032d841
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -29,4 +29,4 @@ model.compile(optimizer='adam', loss='mse')
29
  model.fit(X_train_scaled, y_train, epochs=50, batch_size=32, validation_data=(X_test_scaled, y_test))
30
 
31
  # Save the trained model
32
- model.save('emotion_model')
 
29
  model.fit(X_train_scaled, y_train, epochs=50, batch_size=32, validation_data=(X_test_scaled, y_test))
30
 
31
  # Save the trained model
32
+ model.save('models/emotion_model.h5')