Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import jax
 import jax.numpy as jnp
-from diffusers import FlaxStableDiffusionPipeline
+from diffusers import FlaxPNDMScheduler, FlaxStableDiffusionPipeline
 from flax.jax_utils import replicate
 from flax.training.common_utils import shard
 from share_btn import community_icon_html, loading_icon_html, share_js