Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
@@ -19,8 +19,8 @@ st.caption("Running entirely on CPU using Hugging Face Transformers")
|
|
19 |
# Load the model and tokenizer
|
20 |
@st.cache_resource
|
21 |
def load_model():
|
22 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
23 |
-
model = AutoModelForCausalLM.from_pretrained("
|
24 |
return tokenizer, model
|
25 |
|
26 |
|
|
|
19 |
# Load the model and tokenizer
|
20 |
@st.cache_resource
|
21 |
def load_model():
|
22 |
+
tokenizer = AutoTokenizer.from_pretrained("codellama/CodeLlama-7b-hf")
|
23 |
+
model = AutoModelForCausalLM.from_pretrained("codellama/CodeLlama-7b-hf")
|
24 |
return tokenizer, model
|
25 |
|
26 |
|