Update app.py

app.py

@@ -154,6 +154,8 @@ class PipelineManager:
             return True, random.uniform(0.6, 0.8)
         
         return False, random.uniform(0.1, 0.3)
+    
+    def load_models(self) -> bool:
         """Load models with enhanced error handling and memory optimization"""
         if self.model_loaded:
             return True
@@ -713,33 +715,4 @@ def create_interface():
         
         img_generate_btn.click(
             fn=generate_img2img,
-            inputs=[img_input, img_prompt, img_negative, img_steps, img_guidance,
-                   img_strength, img_seed, img_quality],
-            outputs=[img_output_image, img_download_file, img_info],
-            show_progress=True
-        )
-        
-        # Example prompt buttons
-        txt_example_btn.click(fn=get_random_prompt, outputs=[txt_prompt])
-        img_example_btn.click(fn=get_random_prompt, outputs=[img_prompt])
-        
-        # Clear buttons
-        txt_clear_btn.click(lambda: "", outputs=[txt_prompt])
-        img_clear_btn.click(lambda: "", outputs=[img_prompt])
-    
-    return demo
-
-# Initialize and launch
-if __name__ == "__main__":
-    logger.info(f"🚀 Initializing CyberRealistic Pony Generator on {DEVICE}")
-    logger.info(f"📱 PyTorch version: {torch.__version__}")
-    logger.info(f"🛡️ NSFW Content Filter: Enabled")
-    
-    demo = create_interface()
-    demo.queue(max_size=20)  # Enable queuing for better UX
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        show_error=True,
-        share=False  # Set to True if you want a public link
-    )
+            inputs=[img_input,