Senasu commited on
Commit
cc05458
·
verified ·
1 Parent(s): 2d811c7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ if file is not None:
65
  cols = st.columns(3) # 3 images per row
66
 
67
  for i, idx in enumerate(indices[0]):
68
- product_path = os.path.join('images', filenames[idx])
69
 
70
  st.write(product_path)
71
  product_image = Image.open(product_path)
 
65
  cols = st.columns(3) # 3 images per row
66
 
67
  for i, idx in enumerate(indices[0]):
68
+ product_path = filenames[idx] if filenames[idx].startswith('images/') else os.path.join('images', filenames[idx])
69
 
70
  st.write(product_path)
71
  product_image = Image.open(product_path)