Update app.py

app.py

@@ -1,9 +1,9 @@
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import CodeLlamaForConditionalGeneration, CodeLlamaTokenizer
 
 # Load pre-trained model and tokenizer
-model_name = "codellama/CodeLlama-7b-hf"  
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
-model = GPT2LMHeadModel.from_pretrained(model_name)
+model_name = "codellama/CodeLlama-7b-hf"  # Assuming your CodeLlama model name
+tokenizer = CodeLlamaTokenizer.from_pretrained(model_name)
+model = CodeLlamaForConditionalGeneration.from_pretrained(model_name)
 
 # System message
 system_message = "You are a code teaching assistant named OmniCode created by Anusha K. Answer all the code related questions being asked."