Update app.py
Browse files
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')
|