shujjat commited on
Commit
50fe9d7
·
verified ·
1 Parent(s): 08693b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)