cocoat commited on
Commit
3b111f1
·
verified ·
1 Parent(s): 52dc2ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -12
app.py CHANGED
@@ -3,7 +3,22 @@ import torch
3
  import random
4
  import numpy as np
5
  import datetime
 
 
 
 
6
  import os
 
 
 
 
 
 
 
 
 
 
 
7
  from PIL import Image
8
  from diffusers import (
9
  StableDiffusionXLPipeline,
@@ -39,18 +54,6 @@ euler_scheduler = EulerAncestralDiscreteScheduler.from_config(
39
  dpm_scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
40
  pipe.scheduler = euler_scheduler
41
 
42
- # 履歴保存
43
- from huggingface_hub import login
44
- import os
45
-
46
- # 環境変数 HF_TOKEN が設定されていることを確認し、login() を実行
47
- if "HF_TOKEN" in os.environ:
48
- login(token=os.environ["HF_TOKEN"])
49
- else:
50
- print("警告: HF_TOKEN 環境変数が設定されていません。")
51
- history = []
52
- api = HfApi()
53
-
54
  # 画像をアップロードするリポジトリID
55
  HF_REPO_ID = "cocoat/images"
56
 
 
3
  import random
4
  import numpy as np
5
  import datetime
6
+
7
+ # 履歴保存
8
+ from huggingface_hub import HfApi
9
+ from huggingface_hub import login
10
  import os
11
+ # HF_TOKEN 環境変数からトークンを明示的に読み込む
12
+ hf_token_value = os.getenv("HF_TOKEN")
13
+
14
+ if hf_token_value:
15
+ api = HfApi(token=hf_token_value)
16
+ print("デバッグ情報: Hugging Faceトークンを環境変数から明示的に読み込みました。")
17
+ else:
18
+ # トークンが設定されていない場合の警告と代替処理
19
+ print("エラー: Hugging Faceトークン (HF_TOKEN) が環境変数に設定されていません。アップロード機能は動作しない可能性があります。")
20
+ api = HfApi() # トークンなしで初期化(アップロードは失敗します)
21
+
22
  from PIL import Image
23
  from diffusers import (
24
  StableDiffusionXLPipeline,
 
54
  dpm_scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
55
  pipe.scheduler = euler_scheduler
56
 
 
 
 
 
 
 
 
 
 
 
 
 
57
  # 画像をアップロードするリポジトリID
58
  HF_REPO_ID = "cocoat/images"
59