uripper commited on
Commit
8adddfa
·
1 Parent(s): 0fe7e20

Update app.py

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