Update app.py

app.py

@@ -22,7 +22,14 @@ def hermes_model():
     model = AutoModelForCausalLM.from_pretrained("TheBloke/CapybaraHermes-2.5-Mistral-7B-AWQ", low_cpu_mem_usage=True, device_map="auto")
     return model, tokenizer
 
-model, tokenizer = hermes_model()
+
+
+def blender_model():
+    tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+    model = AutoModelForCausalLM.from_pretrained("facebook/blenderbot-400M-distill", low_cpu_mem_usage=True, device_map="auto")
+    return model, tokenizer
+
+model, tokenizer = blender_model()
 
 # Function to generate a response from the model
 def chat_response(msg_prompt: str) -> str: