Update app.py

app.py

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