Update app.py
Browse files
app.py
CHANGED
@@ -26,10 +26,15 @@ import torch
|
|
26 |
from PIL import Image
|
27 |
from huggingface_hub import login as hf_login
|
28 |
|
29 |
-
# ─────────── Login / device ───────────
|
30 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
31 |
-
if HF_TOKEN:
|
32 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
35 |
DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
|
|
|
26 |
from PIL import Image
|
27 |
from huggingface_hub import login as hf_login
|
28 |
|
|
|
29 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
30 |
+
if not HF_TOKEN:
|
31 |
+
raise RuntimeError(
|
32 |
+
"HF_TOKEN이 설정되지 않았습니다. Space Settings → Secrets에서 "
|
33 |
+
"HF_TOKEN=your_read_token 을 등록한 뒤 재시작하세요."
|
34 |
+
)
|
35 |
+
|
36 |
+
from huggingface_hub import login
|
37 |
+
login(token=HF_TOKEN, add_to_git_credential=False)
|
38 |
|
39 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
40 |
DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
|