Adipta commited on
Commit
bc96d0d
1 Parent(s): 331aefd

edit model

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,12 +3,13 @@ import pandas as pd
3
  import joblib
4
 
5
  data = pd.read_csv(r"data_final.csv")
 
6
 
7
  def product_recommender(customer_id):
8
  list_predicted = []
9
 
10
  for id in data['product_id'].unique():
11
- preds = list(algo.predict(customer_id, id))
12
  product_id = preds[1]
13
  product_score = preds[3]
14
 
 
3
  import joblib
4
 
5
  data = pd.read_csv(r"data_final.csv")
6
+ model = joblib.load("KNN_Model.joblib")
7
 
8
  def product_recommender(customer_id):
9
  list_predicted = []
10
 
11
  for id in data['product_id'].unique():
12
+ preds = list(model.predict(customer_id, id))
13
  product_id = preds[1]
14
  product_score = preds[3]
15