Update app.py

app.py

@@ -3,8 +3,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Load pre-trained model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("distil-gpt2")
-tokenizer = AutoTokenizer.from_pretrained("distil-gpt2")
+model = AutoModelForCausalLM.from_pretrained("bert-base-uncased")
+tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
+
+
 
 def chatbot(input_message):
     input_ids = tokenizer.encode(input_message, return_tensors="pt")