Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,6 @@ import tempfile
|
|
| 7 |
import os
|
| 8 |
import spaces
|
| 9 |
|
| 10 |
-
assert torch.cuda.is_available(), "Demo requires a GPU."
|
| 11 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 12 |
print(DEVICE)
|
| 13 |
|
|
|
|
| 7 |
import os
|
| 8 |
import spaces
|
| 9 |
|
|
|
|
| 10 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 11 |
print(DEVICE)
|
| 12 |
|