girishwangikar commited on
Commit
1813daf
·
verified ·
1 Parent(s): 6e55927

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ class MovieRecommender:
32
  """
33
  result = session.run(query, genre=genre)
34
  df = pd.DataFrame([dict(record) for record in result])
 
35
 
36
  # Add rating category column
37
  df['rating_category'] = pd.cut(
 
32
  """
33
  result = session.run(query, genre=genre)
34
  df = pd.DataFrame([dict(record) for record in result])
35
+ df = df.rename(columns={'avgVote': 'vote_rating'})
36
 
37
  # Add rating category column
38
  df['rating_category'] = pd.cut(