Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 hf_token = os.environ.get("HF_HOME", None)
 
 # Load the model and tokenizer
-model_name = "meta-llama/CodeLlama-7b-hf"
+model_name = "meta-llama/CodeLlama-7b-Python-hf"
 model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token, torch_dtype="float16", device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)