Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -152,11 +152,13 @@ DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stab
|
|
| 152 |
# 環境変数 "token" から Hugging Face トークンを取得
|
| 153 |
hf_token = os.getenv("token")
|
| 154 |
|
| 155 |
-
#
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
|
|
|
|
|
|
| 160 |
os.system(cmd)
|
| 161 |
|
| 162 |
print("Done\nStarting Webui...")
|
|
|
|
| 152 |
# 環境変数 "token" から Hugging Face トークンを取得
|
| 153 |
hf_token = os.getenv("token")
|
| 154 |
|
| 155 |
+
# 削除処理をエラーなしで実行する
|
| 156 |
+
model_path = "/home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors"
|
| 157 |
+
if os.path.exists(model_path):
|
| 158 |
+
os.remove(model_path)
|
| 159 |
+
|
| 160 |
+
# wget でファイルをダウンロード
|
| 161 |
+
cmd = f'wget --header="Authorization: Bearer {hf_token}" -O {model_path} "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"'
|
| 162 |
os.system(cmd)
|
| 163 |
|
| 164 |
print("Done\nStarting Webui...")
|