Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ from huggingface_hub import hf_hub_download
|
|
29 |
|
30 |
if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
|
31 |
device0 = torch.device('cuda:0')
|
32 |
-
device1 = torch.device('cuda:
|
33 |
else:
|
34 |
device0 = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
35 |
device1 = device0
|
@@ -349,7 +349,7 @@ def make3d(images):
|
|
349 |
|
350 |
# print(f"Video saved to {video_fpath}")
|
351 |
|
352 |
-
|
353 |
return mesh_fpath, mesh_glb_fpath
|
354 |
# return video_fpath, mesh_fpath, mesh_glb_fpath
|
355 |
|
|
|
29 |
|
30 |
if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
|
31 |
device0 = torch.device('cuda:0')
|
32 |
+
device1 = torch.device('cuda:0')
|
33 |
else:
|
34 |
device0 = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
35 |
device1 = device0
|
|
|
349 |
|
350 |
# print(f"Video saved to {video_fpath}")
|
351 |
|
352 |
+
mesh_fpath, mesh_glb_fpath = make_mesh(mesh_fpath, planes)
|
353 |
return mesh_fpath, mesh_glb_fpath
|
354 |
# return video_fpath, mesh_fpath, mesh_glb_fpath
|
355 |
|