hysts HF staff commited on
Commit
1489344
·
1 Parent(s): 4fb3c5e

Change default number of steps

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -11,7 +11,7 @@ from model import Model
11
  TITLE = '# Anime Face Generation with [Diffusers](https://github.com/huggingface/diffusers)'
12
  DESCRIPTION = '''
13
 
14
- Expected execution time on Hugging Face Spaces: 13s (DDIM, 50 steps), 6s (PNDM, 20 steps), 247s (DDPM, 1000 steps)
15
  '''
16
  FOOTER = '<img id="visitor-badge" src="https://visitor-badge.glitch.me/badge?page_id=hysts.diffusers-anime-faces" alt="visitor badge" />'
17
 
@@ -41,15 +41,13 @@ def update_scheduler_type(name: str) -> dict:
41
  if name == 'PNDM':
42
  minimum = 4
43
  maximum = 100
44
- value = 20
45
  else:
46
  minimum = 1
47
  maximum = 200
48
- value = 50
49
  return gr.Slider.update(visible=visible,
50
  minimum=minimum,
51
  maximum=maximum,
52
- value=value)
53
 
54
 
55
  def main():
@@ -80,7 +78,7 @@ def main():
80
  num_steps = gr.Slider(1,
81
  200,
82
  step=1,
83
- value=50,
84
  label='Number of Steps')
85
  seed = gr.Slider(0,
86
  100000,
 
11
  TITLE = '# Anime Face Generation with [Diffusers](https://github.com/huggingface/diffusers)'
12
  DESCRIPTION = '''
13
 
14
+ Expected execution time on Hugging Face Spaces: 5s (DDIM, 20 steps), 6s (PNDM, 20 steps), 247s (DDPM, 1000 steps)
15
  '''
16
  FOOTER = '<img id="visitor-badge" src="https://visitor-badge.glitch.me/badge?page_id=hysts.diffusers-anime-faces" alt="visitor badge" />'
17
 
 
41
  if name == 'PNDM':
42
  minimum = 4
43
  maximum = 100
 
44
  else:
45
  minimum = 1
46
  maximum = 200
 
47
  return gr.Slider.update(visible=visible,
48
  minimum=minimum,
49
  maximum=maximum,
50
+ value=20)
51
 
52
 
53
  def main():
 
78
  num_steps = gr.Slider(1,
79
  200,
80
  step=1,
81
+ value=20,
82
  label='Number of Steps')
83
  seed = gr.Slider(0,
84
  100000,