HelloSun commited on
Commit
bad89c4
·
verified ·
1 Parent(s): edfa13e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,7 +28,7 @@ pipe.compile()
28
  prompt=""
29
  negative_prompt=""
30
 
31
- def infer(prompt,negative_prompt, num_inference_steps):
32
 
33
  image = pipe(
34
  prompt = prompt,
@@ -36,7 +36,7 @@ def infer(prompt,negative_prompt, num_inference_steps):
36
  width = 512,
37
  height = 512,
38
  guidance_scale=1.0,
39
- num_inference_steps=num_inference_steps,
40
  num_images_per_prompt=1,
41
  ).images[0]
42
 
@@ -86,7 +86,7 @@ with gr.Blocks(css=css) as demo:
86
 
87
  run_button.click(
88
  fn = infer,
89
- inputs = [prompt,negative_prompt,4],
90
  outputs = [result]
91
  )
92
 
 
28
  prompt=""
29
  negative_prompt=""
30
 
31
+ def infer(prompt,negative_prompt):
32
 
33
  image = pipe(
34
  prompt = prompt,
 
36
  width = 512,
37
  height = 512,
38
  guidance_scale=1.0,
39
+ num_inference_steps=8,
40
  num_images_per_prompt=1,
41
  ).images[0]
42
 
 
86
 
87
  run_button.click(
88
  fn = infer,
89
+ inputs = [prompt,negative_prompt],
90
  outputs = [result]
91
  )
92