Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ from models.build import build_model
|
|
| 20 |
from argparse import Namespace
|
| 21 |
import gdown
|
| 22 |
|
| 23 |
-
|
| 24 |
def down_ckpt(model_cfg, ckpt_dir):
|
| 25 |
model_ids = [
|
| 26 |
['src/config/mp3d.yaml', '1o97oAmd-yEP5bQrM0eAWFPLq27FjUDbh'],
|
|
@@ -137,5 +137,5 @@ if __name__ == '__main__':
|
|
| 137 |
['src/demo/other_demo1.png', False, 'mp3d', 'manhattan', ['depth-normal-gradient', '2d-floorplan'], '.gltf', '256'],
|
| 138 |
['src/demo/other_demo2.png', True, 'mp3d', 'manhattan', ['depth-normal-gradient', '2d-floorplan'], '.gltf', '256'],
|
| 139 |
], title='LGT-Net', allow_flagging="never", cache_examples=False, description=description)
|
| 140 |
-
|
| 141 |
demo.launch(debug=True, enable_queue=False)
|
|
|
|
| 20 |
from argparse import Namespace
|
| 21 |
import gdown
|
| 22 |
|
| 23 |
+
|
| 24 |
def down_ckpt(model_cfg, ckpt_dir):
|
| 25 |
model_ids = [
|
| 26 |
['src/config/mp3d.yaml', '1o97oAmd-yEP5bQrM0eAWFPLq27FjUDbh'],
|
|
|
|
| 137 |
['src/demo/other_demo1.png', False, 'mp3d', 'manhattan', ['depth-normal-gradient', '2d-floorplan'], '.gltf', '256'],
|
| 138 |
['src/demo/other_demo2.png', True, 'mp3d', 'manhattan', ['depth-normal-gradient', '2d-floorplan'], '.gltf', '256'],
|
| 139 |
], title='LGT-Net', allow_flagging="never", cache_examples=False, description=description)
|
| 140 |
+
|
| 141 |
demo.launch(debug=True, enable_queue=False)
|