Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,8 @@ import random
|
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
|
|
|
|
|
|
|
| 9 |
|
| 10 |
if torch.cuda.is_available():
|
| 11 |
device = "cuda"
|
|
@@ -17,6 +19,7 @@ else:
|
|
| 17 |
|
| 18 |
# login hf token
|
| 19 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
|
| 6 |
import numpy as np
|
| 7 |
import os
|
| 8 |
|
| 9 |
+
from huggingface_hub import login
|
| 10 |
+
|
| 11 |
|
| 12 |
if torch.cuda.is_available():
|
| 13 |
device = "cuda"
|
|
|
|
| 19 |
|
| 20 |
# login hf token
|
| 21 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 22 |
+
login(token=HF_TOKEN)
|
| 23 |
|
| 24 |
|
| 25 |
MAX_SEED = np.iinfo(np.int32).max
|