Spaces:
Paused
Paused
Lin Chen
commited on
Commit
·
9ff8061
1
Parent(s):
c073f2e
model_name
Browse files
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=
|
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")
|