Update app.py

app.py

@@ -257,11 +257,15 @@ def ui():
             with gr.Row():
                 secret_token = gr.Text(label='Secret Token', max_lines=1)
 
+                # TODO: find a way to use this to filter the dropdown
+                #base_model = gr.Text(label="Base model")
+            
                 base_model_dropdown = gr.Dropdown(
                     label="Select base Dreambooth model (required)",
                     choices=controller.personalized_model_list,
                     interactive=True,
-                    value="realistic2.safetensors"
+                    "cartoon3d.safetensors"
+                    # value="realistic2.safetensors"
                 )
                 base_model_dropdown.change(fn=controller.update_base_model, inputs=[
                                            base_model_dropdown], outputs=[base_model_dropdown])