Upload app.py

app.py

@@ -55,7 +55,7 @@ def change_base_model(repo_id: str, cn_on: bool): # , progress=gr.Progress(track
     global last_model
     global last_cn_on
     try:
-        if (repo_id == last_model and cn_on is last_cn_on) or not is_repo_name(repo_id) or not is_repo_exists(repo_id): yield gr.update(visible=True)
+        if (repo_id == last_model and cn_on is last_cn_on) or not is_repo_name(repo_id) or not is_repo_exists(repo_id): return gr.update(visible=True)
         if cn_on:
             #progress(0, desc=f"Loading model: {repo_id} / Loading ControlNet: {controlnet_model_union_repo}")
             print(f"Loading model: {repo_id} / Loading ControlNet: {controlnet_model_union_repo}")
@@ -81,7 +81,7 @@ def change_base_model(repo_id: str, cn_on: bool): # , progress=gr.Progress(track
     except Exception as e:
         print(f"Model load Error: {e}")
         raise gr.Error(f"Model load Error: {e}")
-    yield gr.update(visible=True)
+    return gr.update(visible=True)
 
 change_base_model.zerogpu = True
 
@@ -194,10 +194,10 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
 def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height,
               lora_scale, lora_json, cn_on, progress=gr.Progress(track_tqdm=True)):
     global pipe
-    global taef1
-    global good_vae
-    global controlnet
-    global controlnet_union
+    #global taef1
+    #global good_vae
+    #global controlnet
+    #global controlnet_union
     if selected_index is None and not is_valid_lora(lora_json):
         gr.Info("LoRA isn't selected.")
     #    raise gr.Error("You must select a LoRA before proceeding.")