artelabsuper commited on
Commit
e3bb30a
·
1 Parent(s): a05fab5

example like othe demo

Browse files
Files changed (4) hide show
  1. app.py +2 -4
  2. demo_imgs/ex1.png +0 -0
  3. demo_imgs/ex2.png +0 -0
  4. demo_imgs/ex3.png +0 -0
app.py CHANGED
@@ -7,9 +7,7 @@ from PIL import Image
7
  from torchvision import transforms
8
  import matplotlib.pyplot as plt
9
  import numpy as np
10
-
11
-
12
-
13
 
14
  # load model
15
  DEVICE='cpu'
@@ -57,7 +55,7 @@ iface = gr.Interface(
57
  # gr.outputs.Textbox(label='Raw output')
58
  ],
59
  examples=[
60
- ["demo_imgs/fake.jpg"] # use real image
61
  ],
62
  title="DTM Estimation",
63
  description="This demo predict a DTM using GLP Depth model. It will scale input image to 512x512 and at the end it will apply a colormap to better visualize the output."
 
7
  from torchvision import transforms
8
  import matplotlib.pyplot as plt
9
  import numpy as np
10
+ import os
 
 
11
 
12
  # load model
13
  DEVICE='cpu'
 
55
  # gr.outputs.Textbox(label='Raw output')
56
  ],
57
  examples=[
58
+ [f"demo_imgs/{name}"] for name in os.listdir('demo_imgs')
59
  ],
60
  title="DTM Estimation",
61
  description="This demo predict a DTM using GLP Depth model. It will scale input image to 512x512 and at the end it will apply a colormap to better visualize the output."
demo_imgs/ex1.png ADDED
demo_imgs/ex2.png ADDED
demo_imgs/ex3.png ADDED