Update app.py

app.py

@@ -21,7 +21,7 @@ class AIImageGeneratorNSFW:
         if self.is_model_loaded:
             return True
         try:
-            logger.info("Cargando modelo base NSFW con LoRA y optimización GPU...")
+            logger.info("Cargando modelo base NSFW con LoRA y optimización...")
             torch_dtype = torch.float16 if self.device == "cuda" else torch.float32
             
             tokenizer_1 = CLIPTokenizer.from_pretrained(self.model_id, subfolder="tokenizer", use_fast=False)
@@ -41,8 +41,8 @@ class AIImageGeneratorNSFW:
                 variant="fp16" if self.device == "cuda" else None
             )
             
-            # Cargar LoRA usando método load_lora (actual en diffusers)
-            self.pipeline.load_lora(self.lora_id, weight=1.0)  # Ajusta peso según convenga
+            # Carga LoRA con método actual
+            self.pipeline.load_lora(self.lora_id, weight=1.0) 
             
             if self.device == "cuda":
                 self.pipeline.enable_model_cpu_offload()
@@ -84,14 +84,11 @@ class AIImageGeneratorNSFW:
                 torch.cuda.empty_cache()
             return None
 
-# Instancia global para Hugging Face Spaces
-nsfw_app = None
-
 def initialize_generator_nsfw():
-    global nsfw_app
-    if nsfw_app is None:
-        nsfw_app = AIImageGeneratorNSFW()
-    return nsfw_app
+    global generator_nsfw
+    if 'generator_nsfw' not in globals():
+        globals()['generator_nsfw'] = AIImageGeneratorNSFW()
+    return globals()['generator_nsfw']
 
 def generate_image_nsfw(prompt, width, height, steps, guidance_scale):
     gen = initialize_generator_nsfw()
@@ -125,4 +122,5 @@ def create_nsfw_interface():
         )
     return iface
 
+# Declarar interfaz global para Hugging Face Spaces
 nsfw_app = create_nsfw_interface()