Update demo/demo.py

demo/demo.py

@@ -1,7 +1,7 @@
 import gradio as gr
-# from transformers import AutoTokenizer, AutoModel
-# from utils_MMD import extract_features  # Adjust the import path
-# from MMD_calculate import mmd_two_sample_baseline  # Adjust the import path
+from transformers import AutoTokenizer, AutoModel
+from utils_MMD import extract_features  # Adjust the import path
+from MMD_calculate import mmd_two_sample_baseline  # Adjust the import path
 
 MINIMUM_TOKENS = 64
 THRESHOLD = 0.5  # Threshold for classification