Bitha commited on
Commit
6da2908
1 Parent(s): 8a33be1

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,8 +60,8 @@ def get_similar_products_cnn(product_id, num_results):
60
  Productids = list(Productids)
61
  doc_id = Productids.index(product_id)
62
  pairwise_dist = pairwise_distances(extracted_features, extracted_features[doc_id].reshape(1,-1))
63
- indices = np.argsort(pairwise_dist.flatten())[0:num_results]
64
- pdists = np.sort(pairwise_dist.flatten())[0:num_results]
65
  st.write("""
66
  #### input item details
67
  """)
 
60
  Productids = list(Productids)
61
  doc_id = Productids.index(product_id)
62
  pairwise_dist = pairwise_distances(extracted_features, extracted_features[doc_id].reshape(1,-1))
63
+ indices = np.argsort(pairwise_dist.flatten())[0:num_results+1]
64
+ pdists = np.sort(pairwise_dist.flatten())[0:num_results+1]
65
  st.write("""
66
  #### input item details
67
  """)