Bitha commited on
Commit
5001730
1 Parent(s): 3f5df2f

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -61,6 +61,9 @@ st.write(filtered_data)
61
  st.write('---')
62
 
63
  def get_similar_products_cnn(product_id, num_results):
 
 
 
64
  if(fashion_df[fashion_df['ProductId']==product_id]['Gender'].values[0]=="Boys"):
65
  extracted_features = boys_extracted_features
66
  Productids = boys_Productids
 
61
  st.write('---')
62
 
63
  def get_similar_products_cnn(product_id, num_results):
64
+ if product_id not in fashion_df['ProductId'].values:
65
+ st.write("Product ID is not valid")
66
+ return
67
  if(fashion_df[fashion_df['ProductId']==product_id]['Gender'].values[0]=="Boys"):
68
  extracted_features = boys_extracted_features
69
  Productids = boys_Productids