HazardSpence commited on
Commit
067a7d6
·
verified ·
1 Parent(s): 1e03fe1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -2,9 +2,11 @@ import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
 
 
 
5
  model_name = "meta-llama/Meta-Llama-3-8B"
6
- tokenizer = AutoTokenizer.from_pretrained(model_name)
7
- model = AutoModelForCausalLM.from_pretrained(model_name)
8
 
9
 
10
  text_input = st.text_area("Enter text:")
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
 
5
+ access_token = st.secrets["HF_ACCESS_TOKEN"]
6
+
7
  model_name = "meta-llama/Meta-Llama-3-8B"
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name, token=access_token)
9
+ model = AutoModelForCausalLM.from_pretrained(model_name, token=access_token)
10
 
11
 
12
  text_input = st.text_area("Enter text:")