Spaces:
Runtime error
Runtime error
naman7415963
commited on
Commit
•
46d2617
1
Parent(s):
0193b08
Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ import joblib
|
|
5 |
kmeans = joblib.load('kmeans_model.joblib')
|
6 |
|
7 |
movies_pred = pd.read_csv("movies_nonnull.csv")
|
|
|
8 |
with open("movies_df.pkl", "rb") as f:
|
9 |
movies_df = pickle.load(f)
|
10 |
|
@@ -15,7 +16,7 @@ def recommend_movies(name):
|
|
15 |
|
16 |
try:
|
17 |
|
18 |
-
idx = movies_pred[movies_pred['title'] == name].index[0]
|
19 |
|
20 |
prediction = kmeans.predict(movies_df.iloc[idx,:-1].to_numpy().reshape(1,-1))
|
21 |
ans = list(movies_pred[movies_df['KmeansCluster']==prediction[0]].index)
|
|
|
5 |
kmeans = joblib.load('kmeans_model.joblib')
|
6 |
|
7 |
movies_pred = pd.read_csv("movies_nonnull.csv")
|
8 |
+
movies_pred['title'] = movies_pred['title'].apply(lambda x: x.lower().replace(")","").replace("(","").strip())
|
9 |
with open("movies_df.pkl", "rb") as f:
|
10 |
movies_df = pickle.load(f)
|
11 |
|
|
|
16 |
|
17 |
try:
|
18 |
|
19 |
+
idx = movies_pred[movies_pred['title'] == name.lower().strip()].index[0]
|
20 |
|
21 |
prediction = kmeans.predict(movies_df.iloc[idx,:-1].to_numpy().reshape(1,-1))
|
22 |
ans = list(movies_pred[movies_df['KmeansCluster']==prediction[0]].index)
|