Update app.py

app.py

@@ -6,8 +6,12 @@ title = "🤖 AI ChatBot"
 description = "A State-of-the-Art Large-scale Pretrained Response generation model (DialoGPT)"
 examples = [["How are you?"]]
 
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+# tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
+# model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+
+# Load model directly
+tokenizer = AutoTokenizer.from_pretrained("PygmalionAI/pygmalion-6b")
+model = AutoModelForCausalLM.from_pretrained("PygmalionAI/pygmalion-6b")
 
 def generate_response(input_text, chat_history=[]):
     # Tokenize the new input sentence