morizon commited on
Commit
1da142f
1 Parent(s): 6d050af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(
47
  model.eval()
48
  conv_mode = "v1"
49
 
50
- @spaces.GPU
51
  @torch.inference_mode()
52
  def inference_fn(
53
  image,
 
47
  model.eval()
48
  conv_mode = "v1"
49
 
50
+ @spaces.GPU(duration=120)
51
  @torch.inference_mode()
52
  def inference_fn(
53
  image,