daniel-dona commited on
Commit
4d2f992
·
verified ·
1 Parent(s): 1483ea1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -40,8 +40,9 @@ def get_image_size(aspect_ratio):
40
  else:
41
  return 640, 640
42
 
43
- @spaces.GPU(duration=60)
44
- def infer_diffusers(
 
45
  prompt,
46
  negative_prompt=" ",
47
  seed=42,
@@ -145,7 +146,7 @@ with gr.Blocks(css=css) as demo:
145
  gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False, cache_mode="lazy")
146
  gr.on(
147
  triggers=[run_button.click, prompt.submit],
148
- fn=infer_diffusers,
149
  inputs=[
150
  prompt,
151
  negative_prompt,
 
40
  else:
41
  return 640, 640
42
 
43
+ @spaces.GPU(duration=45)
44
+
45
+ def infer(
46
  prompt,
47
  negative_prompt=" ",
48
  seed=42,
 
146
  gr.Examples(examples=examples, inputs=[prompt], outputs=[result, seed], fn=infer, cache_examples=False, cache_mode="lazy")
147
  gr.on(
148
  triggers=[run_button.click, prompt.submit],
149
+ fn=infer,
150
  inputs=[
151
  prompt,
152
  negative_prompt,