mbarnig commited on
Commit
5d9c9c6
·
verified ·
1 Parent(s): acc345c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -24,6 +24,7 @@ MAX_IMAGE_SIZE = 2048
24
  pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
25
 
26
  @spaces.GPU(duration=75)
 
27
  def infer(name, pet, background, style, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
28
  if pet == "Kaatz":
29
  intro = "please generate an image of a cat sitting "
@@ -186,7 +187,7 @@ with gr.Blocks(css=css) as demo:
186
  gr.on(
187
  triggers=[run_button.click, prompt.submit],
188
  fn = infer,
189
- inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
190
  outputs = [result, seed]
191
  )
192
 
 
24
  pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
25
 
26
  @spaces.GPU(duration=75)
27
+
28
  def infer(name, pet, background, style, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
29
  if pet == "Kaatz":
30
  intro = "please generate an image of a cat sitting "
 
187
  gr.on(
188
  triggers=[run_button.click, prompt.submit],
189
  fn = infer,
190
+ inputs = [prompt, pet, background, style, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
191
  outputs = [result, seed]
192
  )
193