uripper commited on
Commit
0baf687
1 Parent(s): 2243ad7
Files changed (2) hide show
  1. app.py +4 -5
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import gradio as gr
2
- from diffusers import PNDMPipeline, PNDMScheduler
3
-
4
  def update():
5
  try:
6
  pipeline.to("cuda")
@@ -11,8 +10,8 @@ def update():
11
  im = images[0]
12
  return im
13
 
14
- scheduler = PNDMScheduler(num_train_timesteps=1100)
15
- pipeline = PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
16
 
17
  css = """
18
  .container {
@@ -29,4 +28,4 @@ with block as demo:
29
  out = gr.Image()
30
  btn = gr.Button("Generate")
31
  btn.click(update, inputs=None, outputs=out)
32
- demo.launch()
 
1
  import gradio as gr
2
+ import diffusers
 
3
  def update():
4
  try:
5
  pipeline.to("cuda")
 
10
  im = images[0]
11
  return im
12
 
13
+ scheduler = diffusers.PNDMScheduler(num_train_timesteps=1100)
14
+ pipeline = diffusers.PNDMPipeline.from_pretrained("uripper/GIANNIS", scheduler=scheduler)
15
 
16
  css = """
17
  .container {
 
28
  out = gr.Image()
29
  btn = gr.Button("Generate")
30
  btn.click(update, inputs=None, outputs=out)
31
+ demo.launch()
requirements.txt CHANGED
@@ -1,2 +1,2 @@
1
- diffusers
2
  torch
 
1
+ diffusers==0.7
2
  torch