uripper commited on
Commit
3a25d8f
1 Parent(s): 015bab3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
- from diffusers import PNDMPipeline, PNDMScheduler
3
  import random
4
 
5
 
6
- scheduler = PNDMScheduler(num_train_timesteps=500,beta_schedule="linear")
7
- pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
8
  # num_imgs = len(os.listdir("\stored_images"))
9
  # imgs = os.listdir("\stored_images")
10
 
 
1
  import gradio as gr
2
+ from diffusers import IPNDMPipeline, IPNDMScheduler
3
  import random
4
 
5
 
6
+ scheduler = IPNDMScheduler(num_train_timesteps=500,beta_schedule="linear")
7
+ pipeline = IPNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
8
  # num_imgs = len(os.listdir("\stored_images"))
9
  # imgs = os.listdir("\stored_images")
10