uripper
commited on
Commit
•
0baf687
1
Parent(s):
2243ad7
updates
Browse files- app.py +4 -5
- requirements.txt +1 -1
app.py
CHANGED
@@ -1,6 +1,5 @@
|
|
1 |
import gradio as gr
|
2 |
-
|
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
|