abhlash commited on
Commit
0df4ea2
·
1 Parent(s): eb1d0fd

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,9 +11,9 @@ load_dotenv()
11
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', stream=sys.stdout)
12
 
13
  # Authenticate with Hugging Face
14
- hf_token = os.environ.get("HUGGINGFACE_TOKEN")
15
  if not hf_token:
16
- raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
17
  login(token=hf_token)
18
 
19
  # Load the Llama-3.1-8B model and tokenizer
 
11
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', stream=sys.stdout)
12
 
13
  # Authenticate with Hugging Face
14
+ hf_token = os.environ.get("HUGGING_FACE_TOKEN")
15
  if not hf_token:
16
+ raise ValueError("HUGGING_FACE_TOKEN not found in environment variables")
17
  login(token=hf_token)
18
 
19
  # Load the Llama-3.1-8B model and tokenizer