brendabor commited on
Commit
1b42e45
1 Parent(s): fc4b9c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -7,7 +7,11 @@ import numpy as np
7
 
8
  # Load your models
9
  emotion_model = load_model('lstm_model.h5')
10
- recommender_model = np.load('knn_model.npy', allow_pickle=True)
 
 
 
 
11
 
12
  # Load the tokenizer (if used during training)
13
  # tokenizer = joblib.load('tokenizer.pkl') # Update with actual file name
 
7
 
8
  # Load your models
9
  emotion_model = load_model('lstm_model.h5')
10
+
11
+ # Save the scikit-learn model
12
+ joblib.dump(knn_model, 'knn_model.pkl')
13
+ # Load the model
14
+ recommender_model = joblib.load('knn_model.pkl')
15
 
16
  # Load the tokenizer (if used during training)
17
  # tokenizer = joblib.load('tokenizer.pkl') # Update with actual file name