yakine commited on
Commit
cd42fe2
·
verified ·
1 Parent(s): 15e9cb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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