Update app.py

app.py

@@ -23,8 +23,7 @@ scheduler = EDMEulerScheduler(
     use_flow_sigmas=True,
 )
 
-pipe_14b.scheduler = scheduler
-# pipe_2b.scheduler = scheduler
+pipe_2b.scheduler = scheduler
 
 @spaces.GPU(duration=85)
 def generate_image(prompt,