dataprincess commited on
Commit
8c587b1
·
verified ·
1 Parent(s): d6fba9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def recommend_laptops(age=None, category=None, gender=None, user_id=None, num_re
41
  new_user_data['Predicted_Rating'] = new_user_data.apply(lambda row: model.predict(999, row['Laptop_ID']).est, axis=1)
42
  recommendations = new_user_data.sort_values(by='Predicted_Rating', ascending=False).head(num_recommendations)
43
 
44
- return recommendations.drop(columns='link')
45
 
46
 
47
  import streamlit as st
 
41
  new_user_data['Predicted_Rating'] = new_user_data.apply(lambda row: model.predict(999, row['Laptop_ID']).est, axis=1)
42
  recommendations = new_user_data.sort_values(by='Predicted_Rating', ascending=False).head(num_recommendations)
43
 
44
+ return recommendations
45
 
46
 
47
  import streamlit as st