Spaces:
Runtime error
Runtime error
artelabsuper
commited on
Commit
·
e3bb30a
1
Parent(s):
a05fab5
example like othe demo
Browse files- app.py +2 -4
- demo_imgs/ex1.png +0 -0
- demo_imgs/ex2.png +0 -0
- 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/
|
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
![]() |