Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from diffusers import IPNDMPipeline, IPNDMScheduler
+from diffusers import PNDMPipeline, IPNDMScheduler
 import random
 
 
-scheduler = IPNDMScheduler(num_train_timesteps=500,beta_schedule="linear")
-pipeline = IPNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
+scheduler = IPNDMScheduler(num_train_timesteps=50)
+pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
 # num_imgs = len(os.listdir("\stored_images"))
 # imgs = os.listdir("\stored_images")