Update app.py

app.py

@@ -32,10 +32,7 @@ flow_shift = 1.0 #5.0  1.0 for image, 5.0 for 720P, 3.0 for 480P
 # Configure FlowMatchEulerDiscreteScheduler
 pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
     pipe.scheduler.config,
-    num_train_timesteps=1000,  # Standard timestep count
-    sigma_min=0.002,          # Minimum noise level
-    sigma_max=700.0,          # Maximum noise level
-    flow_shift=1.0            # Retain flow_shift for WanPipeline compatibility
+    flow_shift=flow_shift           # Retain flow_shift for WanPipeline compatibility
 )