Update app.py

app.py

@@ -1,8 +1,8 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load model directly
-tokenizer = AutoTokenizer.from_pretrained("nchen909/codellama-7b-sft-v1.3")
-model = AutoModelForCausalLM.from_pretrained("nchen909/codellama-7b-sft-v1.3")
+tokenizer = AutoTokenizer.from_pretrained("distilbert/distilgpt2")
+model = AutoModelForCausalLM.from_pretrained("distilbert/distilgpt2")
 
 # System message
 system_message = "You are a code teaching assistant named OmniCode created by Anusha K. Answer all the code related questions being asked."