Update server.py

server.py

@@ -37,7 +37,11 @@ def load_models(model_name="CompVis/stable-diffusion-v1-4"):
     global stable_diff_pipe, model
     
     # Load Stable Diffusion model pipeline
-    stable_diff_pipe = StableDiffusionPipeline.from_pretrained(model_name)
+    stable_diff_pipe = StableDiffusionPipeline.from_pretrained(
+        model_name,
+        safety_checker=None,  # Disable safety checker for feature extraction
+        requires_safety_checker=False  # Explicitly indicate we don't need the safety checker
+    )
     stable_diff_pipe.to("cuda" if torch.cuda.is_available() else "cpu")
     
     # Initialize MIDM model