adjust header

app.py

@@ -64,7 +64,7 @@ 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 = {os.path.basename(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("_")[2].capitalize() for garm_file in garm_list_path}
 
 
 def load_header(header_file):
@@ -146,7 +146,7 @@ def run_turbo(person_img, garm_img, category="Top"):
     return None, info
 
 with gr.Blocks(css=tryon_css) as Huhu_Turbo:
-    gr.HTML(load_header("data/header.md"))
+    gr.HTML(load_header("data/header.html"))
     with gr.Row():
         with gr.Column(elem_id = "col-garment"):
             gr.HTML("""

data/{header.md → header.html}

@@ -10,13 +10,14 @@
         <a href="https://huhu.ai/try-on-api/"><img src="https://img.shields.io/badge/API-Huhu%20AI-red?style=plastic"></a>
     </div>
     <br>
-    <div style="display: flex; text-align: center; font-size: 20px; padding-right: 200px; padding-left: 200px;">
+    <div style="display: flex; text-align: center; font-size: 20px; padding-right: 150px; padding-left: 150px;">
         Try the fastest virtual try-on model — generation can be completed in as little as 10 seconds. <br>
         If you like our project please visit our official website for the latest model updates.
     </div>
     <div style="display: flex; text-align: center; font-size: 16px; padding-right: 200px; padding-left: 200px;">
-        You can not only use your own garments but also upload your own model photos for generation. After uploading a garment image, please select the corresponding clothing type. <br>
-        Please ensure that any images you upload (e.g., model photos) do not infringe third-party rights.
+        You can not only use your own garments but also upload your own model photos for generation. <br>
+        After uploading a garment image, please select the corresponding clothing type. <br>
+        <i>Please ensure that any images you upload (e.g., model photos) do not infringe third-party rights.</i>
     </div>
   </div>
 </div>