Axcel1 commited on
Commit
9baa986
·
verified ·
1 Parent(s): a30ccb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -3
app.py CHANGED
@@ -3,8 +3,26 @@ import os
3
  import threading
4
  import time
5
  from pathlib import Path
6
- from huggingface_hub import hf_hub_download
7
- from huggingface_hub import login
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
 
9
  # Try to import llama-cpp-python, fallback to instructions if not available
10
  try:
@@ -21,7 +39,7 @@ model_loaded = False
21
  # HuggingFace repository information
22
  HF_REPO_ID = "Axcel1/MMed-llama-alpaca-Q4_K_M-GGUF"
23
  HF_FILENAME = "mmed-llama-alpaca-q4_k_m.gguf"
24
- hf_token = os.environ.get("HF_TOKEN")
25
 
26
  login(token = hf_token)
27
 
 
3
  import threading
4
  import time
5
  from pathlib import Path
6
+ from huggingface_hub import hf_hub_download,login
7
+
8
+ # Handle HuggingFace authentication
9
+ def setup_hf_auth():
10
+ """Setup HuggingFace authentication if token is available"""
11
+ hf_token = os.environ.get("HF_TOKEN")
12
+ if hf_token and hf_token.strip():
13
+ try:
14
+ login(token=hf_token, add_to_git_credential=False)
15
+ print("✅ Successfully logged in to HuggingFace Hub")
16
+ return True
17
+ except Exception as e:
18
+ print(f"⚠️ HuggingFace login failed: {e}")
19
+ return False
20
+ else:
21
+ print("ℹ️ No HF_TOKEN found, proceeding without authentication")
22
+ return False
23
+
24
+ # Setup authentication on import
25
+ setup_hf_auth()
26
 
27
  # Try to import llama-cpp-python, fallback to instructions if not available
28
  try:
 
39
  # HuggingFace repository information
40
  HF_REPO_ID = "Axcel1/MMed-llama-alpaca-Q4_K_M-GGUF"
41
  HF_FILENAME = "mmed-llama-alpaca-q4_k_m.gguf"
42
+
43
 
44
  login(token = hf_token)
45