Update app.py

app.py

@@ -89,6 +89,7 @@ import logging
 import csv
 import os
 from datetime import datetime
+from datasets import load_dataset, Dataset
 
 # Model and SentencePiece loading
 sp = spm.SentencePieceProcessor()