LTT commited on
Commit
b020274
1 Parent(s): adc9758

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ from src.utils.infer_util import remove_background, resize_foreground, images_to
26
  import tempfile
27
  from huggingface_hub import hf_hub_download
28
 
29
- print(f"GPU: torch.cuda.is_available()")
30
  if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
31
  device0 = torch.device('cuda:0')
32
  device1 = torch.device('cuda:0')
 
26
  import tempfile
27
  from huggingface_hub import hf_hub_download
28
 
29
+ print(f"GPU: {torch.cuda.is_available()}")
30
  if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
31
  device0 = torch.device('cuda:0')
32
  device1 = torch.device('cuda:0')