naman7415963 commited on
Commit
46d2617
1 Parent(s): 0193b08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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)