Upload 2 files

app.py

@@ -221,7 +221,7 @@ with gr.Blocks(theme=gr.themes.Soft(), fill_width=True, css=css) as app:
         triggers=[generate_button.click, prompt.submit],
         fn=change_base_model,
         inputs=[model_name],
-        outputs=None
+        outputs=[result]
     ).success(
         fn=run_lora,
         inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height,
@@ -229,7 +229,7 @@ with gr.Blocks(theme=gr.themes.Soft(), fill_width=True, css=css) as app:
         outputs=[result, seed]
     )
 
-    model_name.change(change_base_model, [model_name], None)
+    model_name.change(change_base_model, [model_name], [result])
 
     gr.on(
         triggers=[lora_search_civitai_submit.click, lora_search_civitai_query.submit],

mod.py

@@ -73,14 +73,17 @@ base_model = models[0]
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
 
-def change_base_model(repo_id: str):
+def change_base_model(repo_id: str, progress=gr.Progress(track_tqdm=True)):
     global pipe
     try:
         if not is_repo_name(repo_id) or not is_repo_exists(repo_id): return
+        progress(0, f"Loading model: {repo_id}")
         clear_cache()
         pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.bfloat16)
+        progress(1, f"Model loaded: {repo_id}")
     except Exception as e:
         print(e)
+    return gr.update(visible=True)
 
 
 def compose_lora_json(lorajson: list[dict], i: int, name: str, scale: float, filename: str, trigger: str):