Upload app.py

app.py

@@ -75,13 +75,14 @@ def get_similar_products_cnn(product_id, num_results):
          #### Top {num_results} Recommended items
          """)
     for i in range(1,len(indices)):
-        rows = fashion_df[['Gender','ImageURL','ProductTitle']].loc[fashion_df['ProductId']==Productids[indices[i]]]
+        rows = fashion_df[['Gender','ImageURL','ProductTitle','SubCategory']].loc[fashion_df['ProductId']==Productids[indices[i]]]
         for indx, row in rows.iterrows():
             #image = Image.open(Image(url=row['ImageURL'], width = 224, height = 224,embed=True))
             image = Image.open(urllib.request.urlopen(row['ImageURL']))
             image = image.resize((224,224))
             st.image(image)
             st.write(f"Gender Class: {row['Gender']}")
+            st.write(f"Sub Category: {row['SubCategory']}")
             st.write(f"Product Title: {row['ProductTitle']}")
             #st.write(f"Euclidean Distance from input image: {pdists[i]}")