Bardi-ya commited on
Commit
80e0b91
·
verified ·
1 Parent(s): a1e7a4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,16 +1,15 @@
1
  import gradio as gr
2
- import pickle
3
  import pandas as pd
4
  import os
5
 
6
  # Paths
7
- WHO = ""
8
  MODEL_DIR = "models"
9
  MOVIE_DATA_PATH = "data/movies_metadata.csv" # adjust to your actual metadata file
10
 
11
  # Load models (choose what you want to demo)
12
  with open(os.path.join(MODEL_DIR, "recommender_svd_mf.pkl"), "rb") as f:
13
- svd_model = pickle.load(f)
14
 
15
  # Load movie metadata
16
  movies_df = pd.read_csv(MOVIE_DATA_PATH) # should include [movieId, title, poster_url, actors]
 
1
  import gradio as gr
2
+ import joblib
3
  import pandas as pd
4
  import os
5
 
6
  # Paths
 
7
  MODEL_DIR = "models"
8
  MOVIE_DATA_PATH = "data/movies_metadata.csv" # adjust to your actual metadata file
9
 
10
  # Load models (choose what you want to demo)
11
  with open(os.path.join(MODEL_DIR, "recommender_svd_mf.pkl"), "rb") as f:
12
+ svd_model = joblib.load(f)
13
 
14
  # Load movie metadata
15
  movies_df = pd.read_csv(MOVIE_DATA_PATH) # should include [movieId, title, poster_url, actors]