Update depth_estimation.py

depth_estimation.py

@@ -1,7 +1,7 @@
 # Importing the requirements
 import numpy as np
-import torch
 from PIL import Image
+import torch
 from transformers import DPTImageProcessor, DPTForDepthEstimation
 
 
@@ -21,6 +21,9 @@ def process_image(image):
     Returns:
         PIL.Image.Image: The formatted depth map as an image.
     """
+    # Check if the image is provided
+    if not image:
+        gr.error("No image provided")
 
     # Preprocess the image for the model
     encoding = feature_extractor(image, return_tensors="pt")