Update app.py

app.py

@@ -4,11 +4,14 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load model directly
-tokenizer = AutoTokenizer.from_pretrained("distilbert/distilgpt2")
-model = AutoModelForCausalLM.from_pretrained("distilbert/distilgpt2")
+tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
+model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
 
 # System message
-system_message = "I am a code teaching assistant named Repeatinfy created by Anusha K. I am here to complete your sentence and repeat it."
+system_message = '''
+I am a code teaching assistant named as OmniCode created 
+by Anusha K. I will answer all the code related questions being asked."
+'''
 
 
 def generate_response(prompt, max_length=150, temperature=1.0):