Update app.py

app.py

@@ -153,44 +153,39 @@ DownLoad(r"https://civitai.com/api/download/models/39164",str(user_home / r"stab
 os.environ["COMMANDLINE_ARGS"] = "--skip-torch-cuda-test --no-half --use-cpu all"
 
 
-# Hugging Face のリポジトリ情報
-HF_REPO = "https://huggingface.co/black-forest-labs/FLUX.1-dev"
-MODEL_NAME = "flux1-dev.safetensors"
+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, MODEL_NAME)
+MODEL_PATH = os.path.join(MODEL_DIR, FILENAME)
 
-# 環境変数から Hugging Face のトークンを取得
+# 環境変数から Hugging Face トークンを取得
 hf_token = os.getenv("token")
 if not hf_token:
     print("Error: Hugging Face token not found.")
     exit(1)
 
-# `git-lfs` がインストールされているか確認
-try:
-    subprocess.run(["git", "lfs", "version"], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-except subprocess.CalledProcessError:
-    print("Error: Git LFS is not installed. Please install it using:")
-    print("      sudo apt install git-lfs  (Ubuntu/Debian)")
-    print("      brew install git-lfs      (Mac)")
-    print("      choco install git-lfs      (Windows)")
-    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)
 
-# `git-lfs` を使ってモデルをダウンロード
-if not os.path.exists(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)
 
-print("Cloning repository using Git LFS...")
-clone_cmd = [
-    "GIT_LFS_SKIP_SMUDGE=1",  # 最初のクローン時にダウンロードをスキップ
-    "git", "clone", HF_REPO, "--depth=1", "--filter=blob:none", MODEL_DIR
-]
-subprocess.run(" ".join(clone_cmd), shell=True, check=True)
-
-# `git-lfs` で LFS ファイルを取得
-print("Fetching model file using Git LFS...")
-subprocess.run(["git", "lfs", "pull"], cwd=MODEL_DIR, check=True)
+# ダウンロード結果をチェック
+if result.returncode != 0:
+    print(f"Error downloading file: {result.stderr}")
+    exit(1)
 
-# モデルのサイズを確認
+# ファイルサイズを確認
 if os.path.exists(MODEL_PATH):
     file_size = os.path.getsize(MODEL_PATH)
     if file_size < 1000:  # 1KB 未満なら異常
@@ -198,11 +193,13 @@ if os.path.exists(MODEL_PATH):
         os.remove(MODEL_PATH)
         exit(1)
     else:
-        print(f"Download successful. File size: {file_size / (1024 * 1024 * 1024):.2f} GB.")
+        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")