lenawilli commited on
Commit
bb7ac26
·
verified ·
1 Parent(s): f82362f

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -3
src/streamlit_app.py CHANGED
@@ -7,12 +7,12 @@ import joblib
7
 
8
  @st.cache_resource
9
  def load_model():
10
- return tf.keras.models.load_model("../recommender_model.keras")
11
 
12
  @st.cache_data
13
  def load_assets():
14
- df_movies = pd.read_csv("../movies.csv")
15
- user_map, movie_map = joblib.load("../encodings.pkl")
16
  return df_movies, user_map, movie_map
17
 
18
  model = load_model()
 
7
 
8
  @st.cache_resource
9
  def load_model():
10
+ return tf.keras.models.load_model("recommender_model.keras")
11
 
12
  @st.cache_data
13
  def load_assets():
14
+ df_movies = pd.read_csv("movies.csv")
15
+ user_map, movie_map = joblib.load("encodings.pkl")
16
  return df_movies, user_map, movie_map
17
 
18
  model = load_model()