Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,11 +19,10 @@ st.caption("Running entirely on CPU using Hugging Face Transformers")
 # Load the model and tokenizer
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("codellama/CodeLlama-7b-hf")
-    model = AutoModelForCausalLM.from_pretrained("codellama/CodeLlama-7b-hf")
+    tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-Coder-1.3B-base")
+    model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-Coder-1.3B-base")
     return tokenizer, model
 
-
 tokenizer, model = load_model()
 
 # Prompt input