change
Browse files
app.py
CHANGED
|
@@ -26,7 +26,6 @@ import gc
|
|
| 26 |
from datetime import datetime
|
| 27 |
from huggingface_hub import snapshot_download
|
| 28 |
log = logging.getLogger()
|
| 29 |
-
|
| 30 |
device = "cpu"
|
| 31 |
if torch.cuda.is_available():
|
| 32 |
device = "cuda"
|
|
|
|
| 26 |
from datetime import datetime
|
| 27 |
from huggingface_hub import snapshot_download
|
| 28 |
log = logging.getLogger()
|
|
|
|
| 29 |
device = "cpu"
|
| 30 |
if torch.cuda.is_available():
|
| 31 |
device = "cuda"
|