LTT commited on
Commit
ed1060c
1 Parent(s): d0b104b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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:1')
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
- 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
 
 
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