Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 import numpy as np
 from datasets import load_dataset
+from transformers import AutoTokenizer
 import matplotlib.pyplot as plt
 import seaborn as sns
 
 dataset = load_dataset('dar-tau/grammar-attention-maps-opt-350m')['train']
+tokenizer = AutoTokenizer.from_pretrained('facebook/opt-350m', add_prefix_space=True)
 
 def analyze_sentence(index):
     row = dataset[index]