Update app.py

app.py

@@ -146,6 +146,15 @@ def load_and_prepare_model():
     print(f'init noise scale: {pipe.scheduler.init_noise_sigma}')
     pipe.watermark=None
     pipe.safety_checker=None  
+    
+            # Freeze vae and unet
+    pipe.vae.requires_grad_(False)
+    pipe.unet.requires_grad_(False)
+    pipe.text_encoder.requires_grad_(False)
+    pipe.unet.eval()
+    pipe.vae.eval()
+    pipe.text_encoder.eval()
+
     return pipe
     
 # Preload and compile both models