Update app.py

app.py

@@ -1,7 +1,7 @@
 import spaces
 import gradio as gr
 import torch
-from diffusers import FluxPipeline, FluxTransformer2DModel, EulerDiscreteScheduler
+from diffusers import FluxPipeline, FluxTransformer2DModel, FlowMatchEulerDiscreteScheduler
 from huggingface_hub import hf_hub_download
 from PIL import Image
 import requests
@@ -32,7 +32,7 @@ if torch.cuda.is_available():
         model, 
         transformer=transformer,
         torch_dtype=torch.bfloat16)
-    pipe.scheduler = EulerDiscreteScheduler.from_config(
+    pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
         pipe.scheduler.config, use_beta_sigmas=True
     )
     pipe.to("cuda")