Add optional model preloading configuration

app.py

@@ -13,6 +13,7 @@ login(token=hf_token)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
+PRELOAD_MODELS = False  # Easy switch for preloading
 
 _text_gen_pipeline = None
 _image_gen_pipeline = None
@@ -21,6 +22,7 @@ _image_gen_pipeline = None
 def get_image_gen_pipeline():
     global _image_gen_pipeline
     if _image_gen_pipeline is None:
+        print("Loading image generation model on first use...")  # Optional debug message
         try:
             device = "cuda" if torch.cuda.is_available() else "cpu"
             dtype = torch.bfloat16
@@ -180,16 +182,14 @@ def preload_models():
         return False
 
 def create_interface():
-    # Préchargement des modèles
-    models_loaded = preload_models()
-    
-    if not models_loaded:
-        model_status = "⚠️ Erreur lors du chargement des modèles"
+    # Modify the preloading logic
+    if PRELOAD_MODELS:
+        models_loaded = preload_models()
+        model_status = "✅ Modèles chargés avec succès!" if models_loaded else "⚠️ Erreur lors du chargement des modèles"
     else:
-        model_status = "✅ Modèles chargés avec succès!"
+        model_status = "ℹ️ Modèles seront chargés à la demande"
 
     with gr.Blocks(css=css) as demo:
-        
         info = gr.Info(model_status)
 
         with gr.Column(elem_id="col-container"):