Update app.py

app.py

@@ -1,10 +1,8 @@
 import os
 from huggingface_hub import hf_hub_download
 
-# تنظیم مسیرهای کش (برای اطمینان)
+# تنظیم مسیرهای کش
 os.environ["HF_HOME"] = "/home/user/app/hf_cache"
-os.environ["HUGGINGFACE_HUB_CACHE"] = "/home/user/app/hf_cache"
-os.environ["TRANSFORMERS_CACHE"] = "/home/user/app/hf_cache"
 
 # مسیر CelebAMask-HQ
 celeb_path = "/home/user/app/huggingface_models/CelebAMask-HQ"
@@ -13,30 +11,15 @@ face_parsing_path = os.path.join(celeb_path, "face_parsing")
 print("[Info] PYTHONPATH:", os.environ.get("PYTHONPATH"))
 print("[Info] CelebAMask-HQ path exists:", os.path.exists(celeb_path))
 print("[Info] face_parsing folder exists:", os.path.exists(face_parsing_path))
-print("[Info] HF_HOME:", os.environ.get("HF_HOME"))
-print("[Info] Cache directory exists:", os.path.exists("/home/user/app/hf_cache"))
-print("[Info] Cache directory writable:", os.access("/home/user/app/hf_cache", os.W_OK))
 
+# دانلود مدل از Hugging Face
 try:
-    # دانلود مدل از Hugging Face
     model_path = hf_hub_download(
         repo_id="public-data/CelebAMask-HQ-Face-Parsing",
-        filename="model.pth",  # احتمالاً نام فایل درست این است
-        cache_dir="/home/user/app/hf_cache",
-        force_download=False,
-        resume_download=True
+        filename="models/model.pth",
+        cache_dir="/home/user/app/hf_cache"
     )
     print("[Success] Model downloaded to:", model_path)
     
 except Exception as e:
-    print("[Error] Failed to download model:", str(e))
-    # تلاش با نام فایل متفاوت
-    try:
-        model_path = hf_hub_download(
-            repo_id="public-data/CelebAMask-HQ-Face-Parsing",
-            filename="models/model.pth",
-            cache_dir="/home/user/app/hf_cache"
-        )
-        print("[Success] Model downloaded to:", model_path)
-    except Exception as e2:
-        print("[Error] Also failed with alternative filename:", str(e2))
+    print("[Error] Failed to download model:", str(e))