Update app.py

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoModelForCausalLM
 import torch
+from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
 
 # Load the model and tokenizer
 model_name = "GRMenon/mental-health-mistral-7b-instructv0.2-finetuned-V2"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = MistralTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 def get_bot_response(user_input):