preload models

app.py

@@ -4,6 +4,7 @@ import sys
 from typing import Sequence, Mapping, Any, Union
 import torch
 import gradio as gr
+from comfy import model_management
 import spaces
 
 from huggingface_hub import hf_hub_download
@@ -267,6 +268,19 @@ saveimage = NODE_CLASS_MAPPINGS["SaveImage"]()
 imageupscalewithmodel = NODE_CLASS_MAPPINGS["ImageUpscaleWithModel"]()
 
 
+model_loaders = [
+    checkpointloadersimple_50,
+    controlnetloader_73,
+    florence2modelloader_204,
+    checkpointloadersimple_319,
+    loraloader_338,
+    loraloader_353,
+    controlnetloader_389,
+    faceanalysismodels_506,
+    upscalemodelloader_835,
+    layerutility_imageblend_v2
+]
+model_management.load_models_gpu(model_loaders)
 
 @spaces.GPU(duration=60)
 def generate_image(model_image, hairstyle_template_image):