bug fix

app.py

@@ -13,12 +13,11 @@ garm_list_path = [os.path.join(example_path, "garment", garm) for garm in garm_l
 person_list = os.listdir(os.path.join(example_path,"person"))
 person_list_path = [os.path.join(example_path, "person", person) for person in person_list]
 
-garm_img_category_mapping = {garm_file: os.path.basename(garm_file).split("_")[1].capitalize() for garm_file in garm_list_path}
+garm_img_category_mapping = {os.path.basename(garm_file): os.path.basename(garm_file).split("_")[1].capitalize() for garm_file in garm_list_path}
 
 
 def update_category(selected_garm_file):
-    selected_category = garm_img_category_mapping.get(selected_garm_file, None)
-    print(selected_garm_file, selected_category)
+    selected_category = garm_img_category_mapping.get(os.path.basename(selected_garm_file), "Fullbody")
     return gr.update(value=selected_category)
 
 def call_tryon_api(person_file, garm_file, category, model_type='SD_V1'):