Preload default model

app.py

@@ -12,8 +12,6 @@ models = [
 ]
 DEFAULT_MODEL = "stabilityai/stable-diffusion-xl-refiner-1.0"
 
-loaded_models: set[str] = set()
-
 
 def load_pipeline(model):
     pipeline_type = (
@@ -25,6 +23,10 @@ def load_pipeline(model):
     return pipeline_type.from_pretrained(model)
 
 
+load_pipeline(DEFAULT_MODEL).to("cuda")
+loaded_models = {DEFAULT_MODEL}
+
+
 def generate_image(
     model: str,
     prompt: str,