Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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) #
|
| 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")
|
| 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",
|