ssoxye commited on
Commit
809739f
·
1 Parent(s): b5310ec

update preprocess

Browse files
Files changed (1) hide show
  1. preprocess/simple_extractor.py +2 -2
preprocess/simple_extractor.py CHANGED
@@ -373,9 +373,9 @@ def _select_palette_by_category(category: str):
373
  """
374
  category별 palette 선택 로직 (명시적 규칙)
375
  """
376
- if category == "Upper-body":
377
  return get_palette(4)
378
- elif category == "Lower-body":
379
  return get_palette2(4)
380
  elif category == "Dress":
381
  return get_palette(7)
 
373
  """
374
  category별 palette 선택 로직 (명시적 규칙)
375
  """
376
+ if category == "Upper-cloth":
377
  return get_palette(4)
378
+ elif category == "Bottom":
379
  return get_palette2(4)
380
  elif category == "Dress":
381
  return get_palette(7)