yuragoithf commited on
Commit
d9b378b
·
1 Parent(s): a2715eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,7 +4,6 @@ import torch
4
  import numpy as np
5
  from PIL import Image
6
 
7
- torch.hub.download_url_to_file('http://images.cocodataset.org/val2017/000000039769.jpg', 'cats.jpg')
8
  feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
9
  model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
10
 
@@ -33,8 +32,7 @@ def process_image(image):
33
 
34
  title = "Depth Estimation"
35
  description = "Upload an image and get the depth visualization"
36
- examples =[['cats.jpg']]
37
- # examples =[['house.jpg'], ['plane.webp'], ['room.webp']]
38
 
39
  iface = gr.Interface(fn=process_image,
40
  inputs=gr.inputs.Image(type="pil"),
 
4
  import numpy as np
5
  from PIL import Image
6
 
 
7
  feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
8
  model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
9
 
 
32
 
33
  title = "Depth Estimation"
34
  description = "Upload an image and get the depth visualization"
35
+ examples =[['house.jpg'], ['plane.webp'], ['room.webp']]
 
36
 
37
  iface = gr.Interface(fn=process_image,
38
  inputs=gr.inputs.Image(type="pil"),