Update app.py

app.py

@@ -6,6 +6,7 @@ import matplotlib.pyplot as plt
 import subprocess
 import spaces
 
+# Run the script to get pretrained models
 subprocess.run(["bash", "get_pretrained_models.sh"])
 
 # Load model and preprocessing transform
@@ -15,7 +16,7 @@ model.eval()
 @spaces.GPU(duration=120)
 def predict_depth(input_image):
     # Preprocess the image
-    result = depth_pro.load_rgb(input_image.name)
+    result = depth_pro.load_rgb(input_image)  # Removed .name
     image = result[0]
     f_px = result[-1]  # Assuming f_px is the last item in the returned tuple
     image = transform(image)