fix embedder ui

IAHispano · Aug 20, 2024 · 2fd4643 · 2fd4643
1 parent 9a00d1c
commit 2fd4643
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/tabs/train/train.py b/tabs/train/train.py
@@ -428,6 +428,24 @@ def train_tab():
                 value="contentvec",
                 interactive=True,
             )
+
+        with gr.Accordion(
+            i18n("Custom Embedder"), open=True, visible=False
+        ) as embedder_custom:
+            embedder_upload_custom = gr.File(
+                label=i18n("Upload Custom Embedder"),
+                type="filepath",
+                interactive=True,
+            )
+            embedder_custom_refresh = gr.Button(i18n("Refresh"))
+            embedder_model_custom = gr.Dropdown(
+                label=i18n("Custom Embedder"),
+                info=i18n("Select the custom embedder to use for the conversion."),
+                choices=sorted(get_embedder_custom_list()),
+                interactive=True,
+                allow_custom_value=True,
+            )
+
         hop_length = gr.Slider(
             1,
             512,
@@ -486,24 +504,6 @@ def train_tab():
                         interactive=False,
                     )
 
-            with gr.Column(visible=False) as embedder_custom:
-                with gr.Accordion(i18n("Custom Embedder"), open=True):
-                    embedder_upload_custom = gr.File(
-                        label=i18n("Upload Custom Embedder"),
-                        type="filepath",
-                        interactive=True,
-                    )
-                    embedder_custom_refresh = gr.Button(i18n("Refresh"))
-                    embedder_model_custom = gr.Dropdown(
-                        label=i18n("Custom Embedder"),
-                        info=i18n(
-                            "Select the custom embedder to use for the conversion."
-                        ),
-                        choices=sorted(get_embedder_custom_list()),
-                        interactive=True,
-                        allow_custom_value=True,
-                    )
-
         extract_output_info = gr.Textbox(
             label=i18n("Output Information"),
             info=i18n("The output information will be displayed here."),