Lin Chen commited on
Commit
9ff8061
·
1 Parent(s): c073f2e

model_name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ hf_api_token = st.secrets["HUGGINGFACE_API_TOKEN"]
8
  model_name = "meta-llama/Meta-Llama-3.1-8B"
9
 
10
  # Create a text generation pipeline
11
- generator = pipeline("text-generation", model=model, token=hf_api_token)
12
 
13
  # Streamlit UI
14
  st.title("LLaMA 3.1-405B Model Text Generation")
 
8
  model_name = "meta-llama/Meta-Llama-3.1-8B"
9
 
10
  # Create a text generation pipeline
11
+ generator = pipeline("text-generation", model=model_name, token=hf_api_token)
12
 
13
  # Streamlit UI
14
  st.title("LLaMA 3.1-405B Model Text Generation")