Update app.py

app.py

@@ -136,6 +136,7 @@ def infer(
     LCM_adapter=None,        # Параметр для выбора типа LCM_adapter
     use_DDIMScheduler=False, # Параметр для включения DDIMScheduler
     use_Tiny_VAE=False,      # Параметр для включения Tiny_VAE
+    Tiny_VAE=None,           # Параметр для выбора типа Tiny_VAE
     progress=gr.Progress(track_tqdm=True)
 ):
 
@@ -879,13 +880,39 @@ with gr.Blocks(css=css) as demo:
 
         # Tiny_VAE -----------------------------------------------------------------------------------------------------
         # Checkbox для Tiny_VAE
-        with gr.Blocks():
-            use_Tiny_VAE = gr.Checkbox(
-                label="Use Tiny_VAE",
-                value=False,
-                interactive=True
-            )
         
+        # with gr.Blocks():
+        #     use_Tiny_VAE = gr.Checkbox(
+        #         label="Use Tiny_VAE",
+        #         value=False,
+        #         interactive=True
+        #     )
+
+
+
+        with gr.Blocks():
+            with gr.Row():
+                use_Tiny_VAE = gr.Checkbox(
+                    label="Use Tiny_VAE",
+                    value=False,
+                    interactive=True
+                )
+
+                Tiny_VAE = gr.Dropdown(
+                    label="Tiny_VAE Selection",
+                    choices=[
+                        "sd-vae-ft-mse",
+                    ],
+                    value="sd-vae-ft-mse",
+                    visible=False,
+                    interactive=True,
+                )
+
+            use_Tiny_VAE.change(
+                fn=lambda x: gr.update(visible=x), 
+                inputs=use_Tiny_VAE,
+                outputs=Tiny_VAE
+            )Tiny_VAE,           # Параметр для выбора типа Tiny_VAE
         # Удаление фона------------------------------------------------------------------------------------------------
         # Checkbox для удаления фона
         with gr.Blocks():
@@ -933,6 +960,7 @@ with gr.Blocks(css=css) as demo:
             LCM_adapter,        # Параметр для выбора типа LCM_adapter
             use_DDIMScheduler,  # Параметр для включения DDIMScheduler
             use_Tiny_VAE,       # Параметр для включения Tiny_VAE
+            Tiny_VAE,           # Параметр для выбора типа Tiny_VAE
         ],
         outputs=[result],
     )