Update app.py

app.py

@@ -4,7 +4,7 @@ import time
 import pandas as pd
 import torch
 
-from datasets import load_dataset
+from datasets import Dataset, load_dataset
 from sentence_transformers import util
 from transformers import AutoTokenizer, AutoModel