bmay commited on
Commit
1417b09
1 Parent(s): c06c742

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import numpy as np
6
  from transformers import AutoModel
7
  from theia.decoding import load_feature_stats, prepare_depth_decoder, prepare_mask_generator, decode_everything
8
 
9
- @spaces.GPU()
10
  def run_theia(image):
11
  theia_model = AutoModel.from_pretrained("theaiinstitute/theia-tiny-patch16-224-cddsv", trust_remote_code=True)
12
  theia_model = theia_model.to('cuda')
 
6
  from transformers import AutoModel
7
  from theia.decoding import load_feature_stats, prepare_depth_decoder, prepare_mask_generator, decode_everything
8
 
9
+ @spaces.GPU(duration=30)
10
  def run_theia(image):
11
  theia_model = AutoModel.from_pretrained("theaiinstitute/theia-tiny-patch16-224-cddsv", trust_remote_code=True)
12
  theia_model = theia_model.to('cuda')