Update app.py

app.py

@@ -5,19 +5,10 @@ import spaces
 import torch
 import os
 import time
-from diffusers import FluxPipeline, FlowMatchEulerDiscreteScheduler
+from diffusers import FluxPipeline
 
 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to("cuda")
-
-
-pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
-    pipe.scheduler.config,
-    shift=3.16,
-    base_shift=0.5,
-    max_shift=1.15
-)
-
 torch.cuda.empty_cache()
 MAX_IMAGE_SIZE = 2048