Update app.py

app.py

@@ -3,14 +3,16 @@ from PIL import Image
 import src.depth_pro as depth_pro
 import numpy as np
 import matplotlib.pyplot as plt
-import os
+import subprocess
+import spaces
 
-os.system("source get_pretrained_models.sh")
+subprocess.run(["bash", "get_pretrained_models.sh"])
 
 # Load model and preprocessing transform
 model, transform = depth_pro.create_model_and_transforms()
 model.eval()
 
+@spaces.GPU(duration=120)
 def predict_depth(input_image):
     # Preprocess the image
     result = depth_pro.load_rgb(input_image.name)