attempt to fix reload errors

app.py

@@ -273,12 +273,17 @@ class Demo:
                             info="Path or huggingface repo id of the base model that this edit was done against"
                         )
 
-                        self.model_dropdown_export = gr.Dropdown(
-                            label="ESD Model",
-                            choices=list(model_map.keys()),
-                            value='Van Gogh',
-                            interactive=True
-                        )
+                        with gr.Row():
+                            self.model_dropdown_export = gr.Dropdown(
+                                label="ESD Model",
+                                choices=list(model_map.keys()),
+                                value='Van Gogh',
+                                interactive=True
+                            )
+                            self.model_reload_button_export = gr.Button(
+                                value="🔄",
+                                interactive=True
+                            )
 
                         self.save_path_input_export = gr.Text(
                             label="Output path",
@@ -312,8 +317,12 @@ class Demo:
             ]
         )
         self.model_reload_button.click(self.reload_models,
-                                       inputs=[self.model_dropdown],
-                                       outputs=[self.model_dropdown])
+                                       inputs=[self.model_dropdown, self.model_dropdown_export],
+                                       outputs=[self.model_dropdown, self.model_dropdown_export])
+
+        self.model_reload_button_export.click(self.reload_models,
+                                       inputs=[self.model_dropdown, self.model_dropdown_export],
+                                       outputs=[self.model_dropdown, self.model_dropdown_export])
         train_event = self.train_button.click(self.train, inputs = [
             self.train_model_input,
             self.train_img_size_input,
@@ -347,14 +356,16 @@ class Demo:
             self.save_path_input_export,
             self.save_half_export
         ],
-        outputs=[self.export_button, self.export_status, self.export_download]
+        outputs=[self.export_button, self.export_status]
         )
 
-    def reload_models(self, model_dropdown):
+    def reload_models(self, model_dropdown, model_dropdown_export):
         current_model_name = model_dropdown
+        current_model_name_export = model_dropdown_export
         populate_global_model_map()
         global model_names_list
-        return [self.model_dropdown.update(choices=model_names_list, value=current_model_name)]
+        return [gr.update(choices=model_names_list, value=current_model_name),
+                gr.update(choices=model_names_list, value=current_model_name_export)]
 
     def cancel_training(self):
         if self.training:
@@ -473,8 +484,7 @@ class Demo:
             diffuser.pipeline.save_pretrained(save_path)
 
         return [gr.update(interactive=True, value='Export'),
-                gr.update(value=f'Done Exporting!'),
-                save_path]
+                gr.update(value=f'Done Exporting! Diffusers folder is at {os.path.realpath(save_path)}.')]
 
 
     def inference(self, prompt, negative_prompt, seed, width, height, model_name, base_repo_id_or_path, pbar = gr.Progress(track_tqdm=True)):