Update app.py

app.py

@@ -1,11 +1,10 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM
+from transformers import AutoModelForCausalLM, AutoTokenizer
 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 = MistralTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 def get_bot_response(user_input):
@@ -36,4 +35,4 @@ def main():
         st.write(f"Chatbot: {bot_response}")
 
 if __name__ == "__main__":
-    main()
+    main()