sitammeur commited on
Commit
eecb0c0
·
verified ·
1 Parent(s): 9d9399e

Update depth_estimation.py

Browse files
Files changed (1) hide show
  1. depth_estimation.py +1 -1
depth_estimation.py CHANGED
@@ -23,7 +23,7 @@ def process_image(image):
23
  """
24
  # Check if the image is provided
25
  if not image:
26
- gr.error("No image provided")
27
 
28
  # Preprocess the image for the model
29
  encoding = feature_extractor(image, return_tensors="pt")
 
23
  """
24
  # Check if the image is provided
25
  if not image:
26
+ raise gr.Error("No image provided")
27
 
28
  # Preprocess the image for the model
29
  encoding = feature_extractor(image, return_tensors="pt")