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')
|