Update app.py

app.py

@@ -152,16 +152,11 @@ DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stab
 # CPUモード設定（Linux向け修正）
 os.environ["COMMANDLINE_ARGS"] = "--skip-torch-cuda-test --no-half --use-cpu all"
 
+from huggingface_hub import hf_hub_download
 
-import os
-import shutil
-import subprocess
-
-# Hugging Face からダウンロードするファイル情報
 REPO_ID = "black-forest-labs/FLUX.1-dev"
 FILENAME = "flux1-dev.safetensors"
 MODEL_DIR = "/home/user/stable-diffusion-webui/models/Stable-diffusion"
-MODEL_PATH = os.path.join(MODEL_DIR, FILENAME)
 
 # 環境変数から Hugging Face トークンを取得
 hf_token = os.getenv("token")
@@ -169,37 +164,22 @@ if not hf_token:
     print("Error: Hugging Face token not found.")
     exit(1)
 
-# 既存のディレクトリを削除（クローン前にディレクトリを空にする）
-if os.path.exists(MODEL_DIR):
-    print(f"Removing existing directory: {MODEL_DIR}")
-    shutil.rmtree(MODEL_DIR)
+# モデルディレクトリ作成（削除は必要に応じて）
 os.makedirs(MODEL_DIR, exist_ok=True)
 
-# Hugging Face からファイルのみをダウンロード
-print(f"Downloading {FILENAME} from {REPO_ID}...")
-download_cmd = f"wget --header='Authorization: Bearer {hf_token}' -O {MODEL_PATH} https://huggingface.co/{REPO_ID}/resolve/main/{FILENAME}"
-result = subprocess.run(download_cmd, shell=True, capture_output=True, text=True)
-
-# ダウンロード結果をチェック
-if result.returncode != 0:
-    print(f"Error downloading file: {result.stderr}")
+# ダウンロード実行
+try:
+    model_path = hf_hub_download(
+        repo_id=REPO_ID,
+        filename=FILENAME,
+        local_dir=MODEL_DIR,
+        token=hf_token,
+    )
+    print(f"Download successful: {model_path}")
+except Exception as e:
+    print(f"Error downloading model: {e}")
     exit(1)
 
-# ファイルサイズを確認
-if os.path.exists(MODEL_PATH):
-    file_size = os.path.getsize(MODEL_PATH)
-    if file_size < 1000:  # 1KB 未満なら異常
-        print(f"Error: {MODEL_PATH} is too small ({file_size} bytes). Download failed.")
-        os.remove(MODEL_PATH)
-        exit(1)
-    else:
-        print(f"Download successful. File size: {file_size / (1024 ** 3):.2f} GB")
-else:
-    print("Error: Model file not found after download.")
-    exit(1)
-
-print("Download completed successfully.")
-
     
 print("Done\nStarting Webui...")
 os.chdir(user_home / r"stable-diffusion-webui")