Update app.py

app.py

@@ -3,6 +3,7 @@ warnings.filterwarnings('ignore')
 
 import streamlit as st
 import pandas as pd
+from datasets import load_dataset
 import numpy as np
 from sklearn.metrics.pairwise import cosine_similarity
 from transformers import AutoTokenizer, AutoModel