Update app.py

app.py

@@ -58,7 +58,7 @@ pipe = DiffusionPipeline.from_pretrained(
     trust_remote_code=True,
     torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 )
-pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", shift=5)
+pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", trust_remote_code = True, shift=5)
 pipe = pipe.to(device)
 
 MAX_SEED = 2**31 - 1