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
|