Update app.py

app.py

@@ -5,15 +5,18 @@ import torch
 torch.backends.cuda.matmul.allow_tf32 = True
 import gc
 
+model_id = "andite/pastel-mix"
+
+
 controlnet = ControlNetModel.from_pretrained("ioclab/connow", torch_dtype=torch.float16, use_safetensors=True)
 
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "andite/anything-v4.0",
+    model_id,
     controlnet=controlnet,
     torch_dtype=torch.float16,
     safety_checker=None,
 )
-
+pipe = pipe.to("cuda")
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
 pipe.enable_xformers_memory_efficient_attention()