Update app.py

app.py

@@ -17,15 +17,6 @@ vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype
 #    model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
 #)
 
-pipe = StableDiffusionXLPipeline.from_single_file(
-    model_url,
-    torch_dtype = torch.float16,
-    variant = "fp16",
-    vae = vae,
-    use_safetensors = True,
-    use_auth_token="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"
-)
-pipe.to("cuda")
 
 css = """
 .btn-green {
@@ -58,18 +49,14 @@ 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 = StableDiffusionXLPipeline.from_single_file(
+            model_url,
+            torch_dtype = torch.float16,
+            variant = "fp16",
+            vae = vae,
+            use_safetensors = True,
+            use_auth_token="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"
+        )
         pipe.to("cuda")
         return pipe
 
@@ -90,5 +77,6 @@ with gr.Blocks(css=css) as demo:
 
     submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
 
+set_base_model(model_url)
 demo.queue(1)
 demo.launch(debug=True)