Change categories

app.py

@@ -154,9 +154,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container") as demo:
     gr.Markdown("Загрузите фото человека и одежды для виртуальной примерки")
 
     clothing_classes = [
-        "фон", "шляпа", "волосы", "сумка", "рюкзак", "верхняя одежда",
-        "рубашка", "футболка", "брюки", "короткие штаны", "платье", 
-        "юбка", "носки", "обувь", "водолазка", "аксессуары", "руки", "ноги"
+        "фон", "шляпа", "волосы", "очки", "верхняя одежда", "юбка", "брюки", "платье", 
+        "ремень", "левая обувь", "правая обувь", "лицо", "левая нога", "правая нога", 
+        "левая рука", "правая рука", "сумка", "шарф"
     ]
 
     with gr.Row():
@@ -166,7 +166,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container") as demo:
             clothing_label = gr.Dropdown(
                 choices=[(f"{i}: {desc}", i) for i, desc in enumerate(clothing_classes)],
                 label="Класс одежды для маски",
-                value=7
+                value=4
             )
 
             outfit_prompt = gr.Textbox(label="Описание образа (опционально)", placeholder="Например: man in casual outfit")
@@ -176,7 +176,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container") as demo:
 
             gr.Examples(
                 examples=[
-                    ["./test/person2.png", "./test/00008_00.jpg", "man in skirt", "black longsleeve", 7]
+                    ["./test/person2.png", "./test/00008_00.jpg", "man in skirt", "black longsleeve", 4]
                 ],
                 inputs=[person_input, cloth_input, outfit_prompt, clothing_prompt, clothing_label],
                 label="Примеры для быстрого тестирования"