DipendraStha commited on
Commit
76f42fd
·
verified ·
1 Parent(s): 0aed920

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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("deepseek-ai/DeepSeek-Coder-1.3B-base")
23
- model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-Coder-1.3B-base")
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