Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
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()
|