Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModel
+from transformers import LlamaTokenizer, AutoModel
 import plotly.graph_objects as go
 
 model_name = "mistralai/Mistral-7B-v0.1"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = LlamaTokenizer.from_pretrained(model_name)
 model = None
 
 # Set pad token to eos token if not defined