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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,7 +12,7 @@ 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]
16
 
17
  def recommend(user_id, top_k=5):
18
  """Generate top-k recommendations using SVD model."""
@@ -52,7 +52,7 @@ demo = gr.Interface(
52
  ],
53
  outputs=[
54
  gr.Textbox(label="Recommended Movies"),
55
- gr.Gallery(label="Posters").style(grid=[3], height="auto"),
56
  gr.Textbox(label="Explanations")
57
  ],
58
  title="Movie Recommender System",
 
12
  svd_model = joblib.load(f)
13
 
14
  # Load movie metadata
15
+ movies_df = pd.read_csv(MOVIE_DATA_PATH, low_memory=False) # add low_memory=False to suppress DtypeWarning
16
 
17
  def recommend(user_id, top_k=5):
18
  """Generate top-k recommendations using SVD model."""
 
52
  ],
53
  outputs=[
54
  gr.Textbox(label="Recommended Movies"),
55
+ gr.Gallery(label="Posters"),
56
  gr.Textbox(label="Explanations")
57
  ],
58
  title="Movie Recommender System",