Added half of model changing code

I copy pasted some stuff on my phone during my 15 min break at work

app.py

@@ -54,6 +54,24 @@ def generate(prompt, neg_prompt, samp_steps, guide_scale, batch_size, seed, heig
     ).images
     return [(img, f"Image {i+1}") for i, img in enumerate(images)]
         
+def set_base_model(base_model_id):
+        del pipe
+        torch.cuda.empty_cache()
+        gc.collect()
+        pipe = load_pipe(base_model_id)
+        return pipe
+
+def load_pipe(base_model_id):
+        model_id = CONTROLNET_MODEL_IDS[task_name]
+        pipe = DiffusionPipeline.from_pretrained(
+            base_model_id,
+            safety_checker=None,
+            require_safety_checker=False,
+            torch_dtype=torch.float16)
+        pipe.scheduler = UniPCMultistepScheduler.from_config(
+            pipe.scheduler.config)
+        pipe.to("cuda")
+        return pipe
 
 with gr.Blocks(css=css) as demo:
     with gr.Column():