Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ if torch.cuda.is_available():
|
|
29 |
|
30 |
# Function
|
31 |
@spaces.GPU()
|
32 |
-
def generate_image(prompt, ckpt):
|
33 |
global loaded
|
34 |
print(prompt, ckpt)
|
35 |
|
|
|
29 |
|
30 |
# Function
|
31 |
@spaces.GPU()
|
32 |
+
def generate_image(prompt, ckpt="4-steps"):
|
33 |
global loaded
|
34 |
print(prompt, ckpt)
|
35 |
|