vilarin commited on
Commit
263daf5
1 Parent(s): fe16630

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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