Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
+import spaces
 import torch
 import numpy as np
 from transformers import AutoModel
 from theia.decoding import load_feature_stats, prepare_depth_decoder, prepare_mask_generator, decode_everything
 
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
-
+@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(device)
+    theia_model = theia_model.to('cuda')
     target_model_names = [
         "google/vit-huge-patch14-224-in21k",
         "facebook/dinov2-large",