Update app.py

app.py

@@ -34,12 +34,19 @@ JS = """function () {
 }"""
 
 
+vae = AutoencoderKL.from_pretrained(
+    repo,
+    subfolder="vae",
+    torch_dtype=torch.float16,
+)
+
 
 
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
-    pipe = StableDiffusion3Pipeline.from_pretrained(repo, torch_dtype=torch.float16).to("cuda")
+    pipe = StableDiffusion3Pipeline.from_pretrained(repo, vae=vae, torch_dtype=torch.float16).to("cuda")
 
+pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(pipe.scheduler.config)
 
 # Function 
 @spaces.GPU()