Added load_dataset import

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from datasets import load_dataset
 from transformers import pipeline, DistilBertForSequenceClassification, DistilBertTokenizerFast, AutoModelForSequenceClassification, AutoTokenizer, TFAutoModelForSequenceClassification
 
 # Options for models from transformers library