Spaces:

theaiinstitute
/

theia

Running on Zero

bmay commited on Aug 1

Commit

c970642

•

1 Parent(s): f7c781f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from theia.decoding import load_feature_stats, prepare_depth_decoder, prepare_ma
 @spaces.GPU
 def run_theia(image):
-    theia_model = AutoModel.from_pretrained("theaiinstitute/theia-base-patch16-224-cdiv", trust_remote_code=True)
     theia_model = theia_model.to('cuda')
     target_model_names = [
         "google/vit-huge-patch14-224-in21k",

 @spaces.GPU
 def run_theia(image):
+    theia_model = AutoModel.from_pretrained("theaiinstitute/theia-tiny-patch16-224-cddsv", trust_remote_code=True)
     theia_model = theia_model.to('cuda')
     target_model_names = [
         "google/vit-huge-patch14-224-in21k",