miaw1419 commited on
Commit
a591475
·
verified ·
1 Parent(s): 0420280

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -273,6 +273,7 @@ def api_fn(api):
273
  gr.Info("Valid API")
274
  valid_api = api
275
 
 
276
  def generate(prompt, vp_pos, vp_neg, slider):
277
  print(f"prompt: {prompt}")
278
  image = pipe(prompt=prompt,
 
273
  gr.Info("Valid API")
274
  valid_api = api
275
 
276
+ @spaces.GPU(duration=360)
277
  def generate(prompt, vp_pos, vp_neg, slider):
278
  print(f"prompt: {prompt}")
279
  image = pipe(prompt=prompt,