Update app.py

app.py

@@ -19,7 +19,12 @@ import random
 from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM
 from diffusers import DiffusionPipeline
-from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
+try:
+    from diffusers import FlowMatchEulerDiscreteScheduler
+except ImportError:
+    from diffusers.schedulers.scheduling_flow_match_euler_discrete import (
+        FlowMatchEulerDiscreteScheduler,
+    )
 
 REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"