brendabor commited on
Commit
a9ac4db
·
1 Parent(s): bed56fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ def load_models():
30
  scaler_cb = StandardScaler()
31
  audio_features_scaled_cb = scaler_cb.fit_transform(audio_features)
32
  audio_features_df_cb = pd.DataFrame(audio_features_scaled_cb, columns=audio_feature_columns)
33
- combined_features_cb = pd.concat([mood_cats, audio_features_df_cb], axis=1)
34
 
35
  # Load the similarity matrix for content-based
36
  similarity_matrix = np.load('similarity_matrix.npy')
@@ -42,7 +42,7 @@ def load_models():
42
  scaler_knn = StandardScaler()
43
  audio_features_scaled_knn = scaler_knn.fit_transform(audio_features)
44
  audio_features_df_knn = pd.DataFrame(audio_features_scaled_knn, columns=audio_feature_columns)
45
- combined_features_knn = pd.concat([mood_cats_df, audio_features_df_knn], axis=1)
46
 
47
  # Load the KNN model
48
  knn = joblib.load('knn_model.joblib')
 
30
  scaler_cb = StandardScaler()
31
  audio_features_scaled_cb = scaler_cb.fit_transform(audio_features)
32
  audio_features_df_cb = pd.DataFrame(audio_features_scaled_cb, columns=audio_feature_columns)
33
+ combined_features = pd.concat([mood_cats, audio_features_df_cb], axis=1)
34
 
35
  # Load the similarity matrix for content-based
36
  similarity_matrix = np.load('similarity_matrix.npy')
 
42
  scaler_knn = StandardScaler()
43
  audio_features_scaled_knn = scaler_knn.fit_transform(audio_features)
44
  audio_features_df_knn = pd.DataFrame(audio_features_scaled_knn, columns=audio_feature_columns)
45
+ combined_features = pd.concat([mood_cats_df, audio_features_df_knn], axis=1)
46
 
47
  # Load the KNN model
48
  knn = joblib.load('knn_model.joblib')