Spaces:
Sleeping
Sleeping
Commit
·
011f128
1
Parent(s):
514ce55
huggingface_token
Browse files
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"
|