zino36 commited on
Commit
d8dcfc2
·
verified ·
1 Parent(s): 6a56c0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -54,13 +54,13 @@ filepath = hf_hub_download(repo_id="depth-anything/Depth-Anything-V2-Metric-Indo
54
  #model.load_state_dict(state_dict)
55
  #model = model.to(DEVICE).eval()
56
 
57
- pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Metric-Indoor-Large-hf", device=DEVICE )
58
 
59
  title = "# Depth Anything V2"
60
  description = """Official demo for **Depth Anything V2**.
61
  Please refer to our [paper](https://arxiv.org/abs/2406.09414), [project page](https://depth-anything-v2.github.io), and [github](https://github.com/DepthAnything/Depth-Anything-V2) for more details."""
62
 
63
- @spaces.GPU
64
  def predict_depth(image):
65
  return pipe(Image.fromarray(image))["depth"]
66
 
@@ -71,7 +71,7 @@ with gr.Blocks(css="") as demo:
71
 
72
  with gr.Row():
73
  input_image = gr.Image(label="Input Image", type='numpy', elem_id='img-display-input')
74
- depth_image_slider = gr.Image(label="Depth Map with Slider View", elem_id='img-display-output', position=0.5)
75
  submit = gr.Button(value="Compute Depth")
76
  gray_depth_file = gr.File(label="Grayscale depth map", elem_id="download")
77
  raw_file = gr.File(label="16-bit raw output (can be considered as disparity)", elem_id="download")
 
54
  #model.load_state_dict(state_dict)
55
  #model = model.to(DEVICE).eval()
56
 
57
+ pipe = pipeline(task="depth-estimation", model="depth-anything/Depth-Anything-V2-Metric-Indoor-Large-hf", device=0) # Use appropriate device
58
 
59
  title = "# Depth Anything V2"
60
  description = """Official demo for **Depth Anything V2**.
61
  Please refer to our [paper](https://arxiv.org/abs/2406.09414), [project page](https://depth-anything-v2.github.io), and [github](https://github.com/DepthAnything/Depth-Anything-V2) for more details."""
62
 
63
+ @gr.Interface
64
  def predict_depth(image):
65
  return pipe(Image.fromarray(image))["depth"]
66
 
 
71
 
72
  with gr.Row():
73
  input_image = gr.Image(label="Input Image", type='numpy', elem_id='img-display-input')
74
+ depth_image_slider = gr.Image(label="Depth Map with Slider View", elem_id='img-display-output')
75
  submit = gr.Button(value="Compute Depth")
76
  gray_depth_file = gr.File(label="Grayscale depth map", elem_id="download")
77
  raw_file = gr.File(label="16-bit raw output (can be considered as disparity)", elem_id="download")