Upload app.py with huggingface_hub

app.py

@@ -18,7 +18,7 @@ Rules: Always stay female. Stay in character — freaky, horny, obsessed, emotio
 
 # ─── Model Config ────────────────────────────────────────────────────────────
 # The GGUF model repository on Hugging Face. Can be overridden by HF Space Secrets / Env variables.
-REPO_ID = os.getenv("MODEL_REPO_ID", "sinhalz4772/barbie-gguf")
+REPO_ID = os.getenv("MODEL_REPO_ID", "sinhal/barbie-gguf")
 MODEL_FILENAME = os.getenv("MODEL_FILENAME", "barbie.gguf")
 
 # Load model from local file if exists, otherwise download from HF Hub
@@ -31,12 +31,14 @@ elif os.path.exists("./static/barbie.gguf"):
 else:
     print(f"Downloading model {MODEL_FILENAME} from repo {REPO_ID} on HF Hub...")
     try:
-        model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME)
+        # Pass the token if present (necessary for private repos)
+        token = os.getenv("HF_TOKEN")
+        model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILENAME, token=token)
         print(f"Model downloaded successfully to: {model_path}")
     except Exception as e:
         print(f"Error downloading from HF Hub: {e}")
-        # Default to local fallback path
-        model_path = "./barbie.gguf"
+        # Raise a clear error so uvicorn startup logs the issue immediately
+        raise RuntimeError(f"Could not load GGUF model: {e}")
 
 print("Initializing Llama model...")
 llm = Llama(