Update app.py

app.py

@@ -8,7 +8,12 @@ from transformers import (
     AutoTokenizer, AutoModelForCausalLM,
     LogitsProcessor, LogitsProcessorList
 )
-from huggingface_hub import snapshot_download
+from huggingface_hub import snapshot_download, login
+
+# ----------------- Hugging Face Authentication -----------------
+HF_TOKEN = "YOUR_HUGGINGFACE_TOKEN_HERE"  # 🔐 Replace with your real token
+login(token=HF_TOKEN)
+print("🔑 Logged into Hugging Face successfully.")
 
 # ----------------- Basic Environment Setup -----------------
 print("🔧 Initializing environment (CPU mode)...")
@@ -31,6 +36,7 @@ else:
 snapshot_download(
     repo_id="m-a-p/xcodec_mini_infer",
     local_dir="./xcodec_mini_infer",
+    token=HF_TOKEN
 )
 
 sys.path.append(os.path.join(os.path.dirname(__file__), 'xcodec_mini_infer'))
@@ -45,14 +51,13 @@ print("🧠 Loading YuE model on CPU (8-bit)...")
 
 try:
     model = AutoModelForCausalLM.from_pretrained(
-    "m-a-p/YuE-s1-1B-mini-en-cot",  # ✅ smaller model
-    torch_dtype=torch.float32,
-    attn_implementation="eager",
-    low_cpu_mem_usage=True,
-    device_map={"": "cpu"}
+        "m-a-p/YuE-s1-1B-mini-en-cot",  # ✅ smaller model for CPU
+        torch_dtype=torch.float32,
+        attn_implementation="eager",
+        low_cpu_mem_usage=True,
+        device_map={"": "cpu"},
+        token=HF_TOKEN
     )
-
-
     model.eval()
     print("✅ YuE model loaded successfully.")
 except Exception as e:
@@ -63,7 +68,7 @@ except Exception as e:
 print("🎧 Loading codec model...")
 
 basic_model_config = './xcodec_mini_infer/final_ckpt/config.yaml'
-resume_path = './xcodec_mini_infer/final_ckpt/ckpt_00100000.pth'  # smaller ckpt
+resume_path = './xcodec_mini_infer/final_ckpt/ckpt_00100000.pth'
 
 mmtokenizer = _MMSentencePieceTokenizer("./mm_tokenizer_v0.2_hf/tokenizer.model")
 codectool = CodecManipulator("xcodec", 0, 1)