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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
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
- 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...")
 
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...")