yongyeol commited on
Commit
7771b9f
·
verified ·
1 Parent(s): d299e75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
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
- hf_login(token=HF_TOKEN, add_to_git_credential=False)
 
 
 
 
 
 
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