Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ import streamlit as st
|
|
7 |
from transformers import pipeline
|
8 |
|
9 |
# Load the model
|
10 |
-
generator = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-8B")
|
11 |
|
12 |
# Create an API route in Streamlit
|
13 |
@st.cache_resource
|
|
|
7 |
from transformers import pipeline
|
8 |
|
9 |
# Load the model
|
10 |
+
generator = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-8B", token= hf_token)
|
11 |
|
12 |
# Create an API route in Streamlit
|
13 |
@st.cache_resource
|