Update app.py

app.py

@@ -7,6 +7,7 @@ import torch.nn.functional as F
 from torch.utils.data import Dataset, DataLoader
 from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
 from sklearn.model_selection import train_test_split
+from sklearn.feature_extraction.text import CountVectorizer # Converts text into a matrix of token counts
 from sklearn.metrics import classification_report, accuracy_score
 import gradio as gr