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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -199,7 +199,7 @@ def clear_comments():
199
  return extract_vp_botton, clear_botton
200
 
201
 
202
- @spaces.GPU(duration=360)
203
  def extract_vp():
204
  if valid_api == "":
205
  vpe_model = AutoModelForVision2Seq.from_pretrained(
@@ -273,7 +273,7 @@ def api_fn(api):
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,
 
199
  return extract_vp_botton, clear_botton
200
 
201
 
202
+ @spaces.GPU(duration=200)
203
  def extract_vp():
204
  if valid_api == "":
205
  vpe_model = AutoModelForVision2Seq.from_pretrained(
 
273
  gr.Info("Valid API")
274
  valid_api = api
275
 
276
+ @spaces.GPU(duration=90)
277
  def generate(prompt, vp_pos, vp_neg, slider):
278
  print(f"prompt: {prompt}")
279
  image = pipe(prompt=prompt,