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
|