Update app.py

app.py

@@ -187,7 +187,7 @@ class PipelineManager:
             # Apply memory optimizations to txt2img pipeline
             self._optimize_pipeline(self.txt2img_pipe)
             
-            # Create img2img pipeline sharing components to save memory
+            # Create img2img pipeline sharing components
             self.img2img_pipe = StableDiffusionXLImg2ImgPipeline(
                 vae=self.txt2img_pipe.vae,
                 text_encoder=self.txt2img_pipe.text_encoder,
@@ -196,8 +196,7 @@ class PipelineManager:
                 tokenizer_2=self.txt2img_pipe.tokenizer_2,
                 unet=self.txt2img_pipe.unet,
                 scheduler=self.txt2img_pipe.scheduler,
-                safety_checker=None, # Disable for faster loading, using custom NSFW check
-                requires_safety_checker=False
+                # Removed safety_checker and requires_safety_checker as they are not valid for this constructor
             )
             
             # Apply memory optimizations to img2img pipeline
@@ -421,7 +420,7 @@ def generate_txt2img(prompt: str, negative_prompt: str, steps: int, guidance_sca
         logger.error(f"Generation error: {e}")
         return None, None, f"❌ Generation failed: {str(e)}"
     finally:
-        pipe_manager.clear_memory() # Ensure memory is cleared even if an error occurs
+        pipe_manager.clear_memory() # Ensure memory is cleared even if an occurs
 
 @spaces.GPU(duration=120) # Increased duration for model loading and generation
 def generate_img2img(input_image: Image.Image, prompt: str, negative_prompt: str,