brendabor commited on
Commit
c4d5407
1 Parent(s): 1dd9cbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -2,14 +2,18 @@ import streamlit as st
2
  from tensorflow.keras.models import load_model
3
  from tensorflow.keras.preprocessing.text import Tokenizer
4
  from tensorflow.keras.preprocessing.sequence import pad_sequences
5
- #import joblib
6
- #pip install joblib
7
-
8
 
9
  # Load your models
10
  emotion_model = load_model('lstm_model.h5')
11
  recommender_model = joblib.load('knn_model.npy')
12
 
 
 
 
 
13
  st.title("Emotion-based Song Recommender")
14
 
15
  # User input for lyrics
 
2
  from tensorflow.keras.models import load_model
3
  from tensorflow.keras.preprocessing.text import Tokenizer
4
  from tensorflow.keras.preprocessing.sequence import pad_sequences
5
+ import joblib
6
+ import pandas as pd
7
+ import numpy as np
8
 
9
  # Load your models
10
  emotion_model = load_model('lstm_model.h5')
11
  recommender_model = joblib.load('knn_model.npy')
12
 
13
+ #load the dataset
14
+ # df = pd.read_csv('path_to_your_dataframe.csv')
15
+
16
+
17
  st.title("Emotion-based Song Recommender")
18
 
19
  # User input for lyrics