Update app.py

app.py

@@ -6,7 +6,7 @@ from config import epochs, batch_size, learning_rate
 from model import tokenizer, multitask_model
 from mtm import MultitaskTrainer, NLPDataCollator, DataLoaderWithTaskname
 import pandas as pd
-# from data_5_LT23 import features_dict,extra_feature_dict
+from datasets import Dataset, DatasetDict
 from data_predict import convert_to_stsb_features,convert_to_features
 
 from huggingface_hub import hf_hub_download,snapshot_download