init project
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ import torchvision.transforms as tvf
|
|
| 39 |
|
| 40 |
|
| 41 |
silent = False
|
| 42 |
-
pe3r = Models('cpu')
|
| 43 |
|
| 44 |
|
| 45 |
def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
|
|
|
|
| 39 |
|
| 40 |
|
| 41 |
silent = False
|
| 42 |
+
pe3r = Models('cuda' if torch.cuda.is_available() else 'cpu')
|
| 43 |
|
| 44 |
|
| 45 |
def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
|