pentarosarium commited on
Commit
8f1d8ab
·
verified ·
1 Parent(s): 8c11454

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from transformers import pipeline
8
  # Load the model (Meta-Llama 3.1 8B)
9
  @st.cache_resource
10
  def load_model():
11
- model = pipeline("text2text-generation", model="meta-llama/Meta-Llama-3.1-8B-Instruct")
12
  return model
13
 
14
  model = load_model()
 
8
  # Load the model (Meta-Llama 3.1 8B)
9
  @st.cache_resource
10
  def load_model():
11
+ model = pipeline("text2text-generation", model="meta-llama/Meta-Llama-3.1-8B-Instruct", use_auth_token=True)
12
  return model
13
 
14
  model = load_model()