Update app.py

app.py

@@ -306,15 +306,15 @@ def format_generation_info(params: Dict[str, Any], generation_time: float) -> st
     return "\n".join(info_lines)
 
 @spaces.GPU(duration=120)  # Increased duration for model loading
-def generate_txt2img(prompt: str, negative_prompt: str, steps: int, guidance_scale: float, 
+def generate_txt2img(prompt: str, negative_prompt: str, steps: int, guidance_scale: float,
                     width: int, height: int, seed: int, add_quality: bool) -> Tuple:
     """Text-to-image generation with enhanced error handling"""
-    
+
     if not prompt.strip():
         return None, None, "❌ Please enter a prompt"
-    
+
     # Lazy load models
-    if not pipe_manager.load_models():
+    if not pipe_manager._load_models(): # <--- Change from load_models() to _load_models()
         return None, None, "❌ Failed to load model. Please try again."
     
     try:
@@ -405,18 +405,18 @@ def generate_txt2img(prompt: str, negative_prompt: str, steps: int, guidance_sca
         pipe_manager.clear_memory()
 
 @spaces.GPU(duration=120)
-def generate_img2img(input_image: Image.Image, prompt: str, negative_prompt: str, 
-                    steps: int, guidance_scale: float, strength: float, seed: int, 
+def generate_img2img(input_image: Image.Image, prompt: str, negative_prompt: str,
+                    steps: int, guidance_scale: float, strength: float, seed: int,
                     add_quality: bool) -> Tuple:
     """Image-to-image generation with enhanced preprocessing"""
-    
+
     if input_image is None:
         return None, None, "❌ Please upload an input image"
-    
+
     if not prompt.strip():
         return None, None, "❌ Please enter a prompt"
-    
-    if not pipe_manager.load_models():
+
+    if not pipe_manager._load_models(): # <--- Change from load_models() to _load_models()
         return None, None, "❌ Failed to load model. Please try again."
     
     try: