Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,7 @@ st.title("Meta LLaMA Text Generation")
|
|
6 |
@st.cache_resource
|
7 |
def load_model():
|
8 |
model_name = "meta-llama/Meta-Llama-3-8B"
|
|
|
9 |
|
10 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
|
11 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=access_token)
|
|
|
6 |
@st.cache_resource
|
7 |
def load_model():
|
8 |
model_name = "meta-llama/Meta-Llama-3-8B"
|
9 |
+
access_token = os.getenv('hf')
|
10 |
|
11 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=access_token)
|
12 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=access_token)
|