finished code to change model

app.py

@@ -13,6 +13,7 @@ model_list = ["stabilityai/stable-diffusion-xl-base-1.0",
 model_url_list = ["stabilityai/stable-diffusion-xl-base-1.0/blob/main/sd_xl_base_1.0.safetensors",
                  "Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
                  "Krebzonide/Sevenof9_v3_sdxl/blob/main/nsfwSevenof9V3_nsfwSevenof9V3.safetensors"]
+current_model = 0
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = None
@@ -51,6 +52,12 @@ def generate(prompt, neg_prompt, samp_steps, guide_scale, batch_size, seed, heig
         
 def set_base_model(base_model_id):
     global pipe
+    global current_model
+    global model_list
+    global model_url_list
+    new_model = model_list.index(base_model_id)
+    if new_model == current_model:
+        return pipe
     del pipe
     torch.cuda.empty_cache()
     gc.collect()
@@ -63,8 +70,8 @@ def set_base_model(base_model_id):
         use_auth_token="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"
     )
     pipe.to("cuda")
+    current_model = new_model
     return pipe
-    
 
 with gr.Blocks(css=css) as demo:
     with gr.Column():
@@ -85,8 +92,8 @@ with gr.Blocks(css=css) as demo:
             change_model_btn = gr.Button("Update Model", elem_classes="btn-green")
 
     submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
-    change_model_btn.click(set_base_model, [model_id], queue = false)
+    change_model_btn.click(set_base_model, [model_id], queue = False)
 
-pipe = set_base_model(model_list[0])
+pipe = set_base_model(model_list[current_model])
 demo.queue(1)
 demo.launch(debug=True)