Update server.py

server.py

@@ -34,7 +34,7 @@ def load_models():
     global stable_diff_pipe, model
     
     # Load Stable Diffusion model pipeline
-    stable_diff_pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4-original")
+    stable_diff_pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
     stable_diff_pipe.to("cuda" if torch.cuda.is_available() else "cpu")
     
     # Initialize MIDM model