soiz1 commited on
Commit
5d4760f
·
verified ·
1 Parent(s): 7fcf588

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -149,10 +149,15 @@ DownLoad(r"https://civitai.com/api/download/models/39885",str(user_home / r"stab
149
  DownLoad(r"https://civitai.com/api/download/models/21065",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"LAS.safetensors")
150
  DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"backlighting.safetensors")
151
 
152
- os.system('export HF_TOKEN=' + os.getenv("token")) # Hugging Faceのアクセストークンを入力
153
- os.system('wget --header="Authorization: Bearer $HF_TOKEN" -O /home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"')
154
-
155
-
 
 
 
 
 
156
 
157
  print("Done\nStarting Webui...")
158
  os.chdir(user_home / r"stable-diffusion-webui")
 
149
  DownLoad(r"https://civitai.com/api/download/models/21065",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"LAS.safetensors")
150
  DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stable-diffusion-webui" / r"extensions" / r"sd-webui-additional-networks" / r"models"/ r"lora"),r"backlighting.safetensors")
151
 
152
+ # 環境変数 "token" から Hugging Face トークンを取得
153
+ hf_token = os.getenv("token")
154
+
155
+ # モデルファイルを削除し、新しくダウンロード
156
+ cmd = f"""
157
+ rm /home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors && \
158
+ wget --header="Authorization: Bearer {hf_token}" -O /home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"
159
+ """
160
+ os.system(cmd)
161
 
162
  print("Done\nStarting Webui...")
163
  os.chdir(user_home / r"stable-diffusion-webui")