Update app.py

app.py

@@ -49,10 +49,11 @@ def load_models():
             pipeline.safety_checker = None
             print("✅ NSFW filter disabled", file=sys.stderr)
         
+        # Alternative: Replace safety checker with dummy function
         # Alternative: Replace safety checker with dummy function
         if hasattr(pipeline, 'run_safety_checker'):
-            def dummy_safety_checker(images, *args, **kwargs):
-                return images, [False] * len(images)
+            def dummy_safety_checker(image, *args, **kwargs):
+                return image, False
             pipeline.run_safety_checker = dummy_safety_checker
             print("✅ Safety checker replaced with dummy", file=sys.stderr)