diff --git a/diann_lib_processing.py b/diann_lib_processing.py
index dbf35f03b41766b4c4a93f8588a5589824c0ffd7..51ba849b5896c1defa87284e239807a95c1d6335 100644
--- a/diann_lib_processing.py
+++ b/diann_lib_processing.py
@@ -72,7 +72,8 @@ if __name__ =='__main__':
 
     model.load_state_dict(torch.load(args.model_weigh, weights_only=True))
 
-    data_test = load_data(data_source=args.dataset_test, batch_size=args.batch_size, length=30, mode=args.split_test,
+    print(args.dataset_test)
+    data_test = load_data(data_source='data/spectral_lib/data_uniprot_base.csv', batch_size=args.batch_size, length=30, mode=args.split_test,
                           seq_col=args.seq_test)
 
     predict(data_test, model, args.output)
\ No newline at end of file