Update app.py

app.py

@@ -8,6 +8,8 @@ from datasets import load_dataset
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer
 import numpy as np
 import pandas as pd 
+from transformers import load_dataset
+
 kp = KeyPhraseTransformer()
 
 @st.cache_resource