HelloSun commited on
Commit
0fde7a3
·
verified ·
1 Parent(s): 19cfd55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ batch_size, num_images, height, width = 1, 1, 512, 512
18
  pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
19
  pipeline.compile()
20
 
21
- def infer(prompt, num_inference_steps):
22
 
23
  image = pipeline(
24
  prompt = prompt,
@@ -93,7 +93,7 @@ with gr.Blocks(css=css) as demo:
93
 
94
  run_button.click(
95
  fn = infer,
96
- inputs = [prompt, num_inference_steps],
97
  outputs = [result]
98
  )
99
 
 
18
  pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
19
  pipeline.compile()
20
 
21
+ def infer(prompt,negative_prompt, num_inference_steps):
22
 
23
  image = pipeline(
24
  prompt = prompt,
 
93
 
94
  run_button.click(
95
  fn = infer,
96
+ inputs = [prompt, negative_prompt,num_inference_steps],
97
  outputs = [result]
98
  )
99