roman-bachmann commited on
Commit
d41a2a0
1 Parent(s): 3222587

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def img_from_path(img_path: str):
67
  img = rgb_transform.postprocess(img_pil).unsqueeze(0)
68
  return img
69
 
70
- @spaces.GPU
71
  def infer(img_path, seed=0, randomize_seed=False, target_modalities=None, top_p=0.8, top_k=0.0):
72
  if randomize_seed:
73
  seed = None
 
67
  img = rgb_transform.postprocess(img_pil).unsqueeze(0)
68
  return img
69
 
70
+ @spaces.GPU(duration=80)
71
  def infer(img_path, seed=0, randomize_seed=False, target_modalities=None, top_p=0.8, top_k=0.0):
72
  if randomize_seed:
73
  seed = None