leedoming commited on
Commit
3ddea50
ยท
verified ยท
1 Parent(s): 70e228b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -88,7 +88,9 @@ if api_key:
88
  database_info = []
89
  for item in data:
90
  image_url = item['์ด๋ฏธ์ง€ ๋งํฌ'][0]
91
- image_path = f"temp_{item['\ufeff์ƒํ’ˆ ID']}.jpg"
 
 
92
  response = requests.get(image_url)
93
  with open(image_path, 'wb') as f:
94
  f.write(response.content)
@@ -98,7 +100,7 @@ if api_key:
98
 
99
  database_embeddings.append(embedding)
100
  database_info.append({
101
- 'id': item['\ufeff์ƒํ’ˆ ID'],
102
  'category': item['์นดํ…Œ๊ณ ๋ฆฌ'],
103
  'brand': item['๋ธŒ๋žœ๋“œ๋ช…'],
104
  'name': item['์ œํ’ˆ๋ช…'],
 
88
  database_info = []
89
  for item in data:
90
  image_url = item['์ด๋ฏธ์ง€ ๋งํฌ'][0]
91
+ # '\ufeff์ƒํ’ˆ ID' ๋Œ€์‹  '์ƒํ’ˆ ID'๋ฅผ ์‚ฌ์šฉํ•˜๊ฑฐ๋‚˜, ๋‹ค์Œ๊ณผ ๊ฐ™์ด ์ˆ˜์ •
92
+ product_id = item.get('\ufeff์ƒํ’ˆ ID') or item.get('์ƒํ’ˆ ID')
93
+ image_path = "temp_{}.jpg".format(product_id)
94
  response = requests.get(image_url)
95
  with open(image_path, 'wb') as f:
96
  f.write(response.content)
 
100
 
101
  database_embeddings.append(embedding)
102
  database_info.append({
103
+ 'id': product_id,
104
  'category': item['์นดํ…Œ๊ณ ๋ฆฌ'],
105
  'brand': item['๋ธŒ๋žœ๋“œ๋ช…'],
106
  'name': item['์ œํ’ˆ๋ช…'],