Update app.py

app.py

@@ -57,11 +57,6 @@ repo_large_path = snapshot_download(repo_id="stabilityai/stable-diffusion-3-medi
 repo_large_transformer_path = SD3Transformer2DModel.from_pretrained(repo_large_path, subfolder="transformer", torch_dtype=torch.float16)
 repo_large_transformer = PeftModel.from_pretrained(repo_large_transformer_path, "jasperai/flash-sd3")
 
-pipe.scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(
-  repo_large_path,
-  subfolder="scheduler",
-)
-
 repo_customs = {
     "Default": None, #repo_default,
     "Realistic": None, #StableDiffusionXLPipeline.from_pretrained("ehristoforu/Visionix-alpha", torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
@@ -72,6 +67,11 @@ repo_customs = {
         tokenizer_3=None),
 }
 
+repo_customs["Large"].scheduler = FlashFlowMatchEulerDiscreteScheduler.from_pretrained(
+  repo_large_path,
+  subfolder="scheduler",
+)
+
 # Functions
 def save_image(img, seed):
     name = f"{seed}-{uuid.uuid4()}.png"