Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ from huggingface_hub import login
|
|
5 |
import torch
|
6 |
import os
|
7 |
hf_token = os.getenv("llama")
|
|
|
8 |
#login(hf_token)
|
9 |
# Model and adapter paths
|
10 |
model_name = "unsloth/llama-3.2-1b-instruct-bnb-4bit" # Base model
|
|
|
5 |
import torch
|
6 |
import os
|
7 |
hf_token = os.getenv("llama")
|
8 |
+
assert hf_token is not None, "Token is missing! Make sure 'llama' is set in the environment."
|
9 |
#login(hf_token)
|
10 |
# Model and adapter paths
|
11 |
model_name = "unsloth/llama-3.2-1b-instruct-bnb-4bit" # Base model
|