beingcognitive commited on
Commit
011f128
·
1 Parent(s): 514ce55

huggingface_token

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -6,8 +6,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
6
  import torch
7
  from huggingface_hub import login
8
 
 
 
 
 
9
  # Authenticate with Hugging Face
10
- login()
11
 
12
  # Load model and tokenizer
13
  model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
6
  import torch
7
  from huggingface_hub import login
8
 
9
+ from dotenv import load_dotenv
10
+ # Load environment variables
11
+ load_dotenv()
12
+
13
  # Authenticate with Hugging Face
14
+ login(token=os.getenv("HUGGINGFACE_TOKEN"))
15
 
16
  # Load model and tokenizer
17
  model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"