zc-huhu commited on
Commit
a59c179
·
1 Parent(s): ebe7ed8
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -15,10 +15,12 @@ person_list_path = [os.path.join(example_path, "person", person) for person in p
15
 
16
  garm_img_category_mapping = {garm_file: os.path.basename(garm_file).split("_")[1].capitalize() for garm_file in garm_list_path}
17
 
 
 
18
 
19
  def update_category(selected_garm_file):
20
  selected_category = garm_img_category_mapping.get(selected_garm_file, None)
21
- # Return an update dict for the Dropdown
22
  return gr.update(value=selected_category)
23
 
24
  def call_tryon_api(person_file, garm_file, category, model_type='SD_V1'):
 
15
 
16
  garm_img_category_mapping = {garm_file: os.path.basename(garm_file).split("_")[1].capitalize() for garm_file in garm_list_path}
17
 
18
+ print(garm_img_category_mapping)
19
+
20
 
21
  def update_category(selected_garm_file):
22
  selected_category = garm_img_category_mapping.get(selected_garm_file, None)
23
+ print(selected_category)
24
  return gr.update(value=selected_category)
25
 
26
  def call_tryon_api(person_file, garm_file, category, model_type='SD_V1'):