Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import dask
|
|
4 |
from dask import delayed
|
5 |
|
6 |
# Load model
|
7 |
-
pipe = DiffusionPipeline.from_pretrained("
|
8 |
|
9 |
def generate_image(prompt, num_inference_steps=50):
|
10 |
"""
|
|
|
4 |
from dask import delayed
|
5 |
|
6 |
# Load model
|
7 |
+
pipe = DiffusionPipeline.from_pretrained("kandinsky-community/kandinsky-2-1")
|
8 |
|
9 |
def generate_image(prompt, num_inference_steps=50):
|
10 |
"""
|