Update app.py

app.py

@@ -3,8 +3,26 @@ import os
 import threading
 import time
 from pathlib import Path
-from huggingface_hub import hf_hub_download
-from huggingface_hub import login
+from huggingface_hub import hf_hub_download,login
+
+# Handle HuggingFace authentication
+def setup_hf_auth():
+    """Setup HuggingFace authentication if token is available"""
+    hf_token = os.environ.get("HF_TOKEN")
+    if hf_token and hf_token.strip():
+        try:
+            login(token=hf_token, add_to_git_credential=False)
+            print("✅ Successfully logged in to HuggingFace Hub")
+            return True
+        except Exception as e:
+            print(f"⚠️ HuggingFace login failed: {e}")
+            return False
+    else:
+        print("ℹ️ No HF_TOKEN found, proceeding without authentication")
+        return False
+
+# Setup authentication on import
+setup_hf_auth()
 
 # Try to import llama-cpp-python, fallback to instructions if not available
 try:
@@ -21,7 +39,7 @@ model_loaded = False
 # HuggingFace repository information
 HF_REPO_ID = "Axcel1/MMed-llama-alpaca-Q4_K_M-GGUF"
 HF_FILENAME = "mmed-llama-alpaca-q4_k_m.gguf"
-hf_token = os.environ.get("HF_TOKEN")
+
 
 login(token = hf_token)