diff --git a/function.py b/function.py
index 62268a0..8ce3dbf 100644
--- a/function.py
+++ b/function.py
@@ -155,6 +155,13 @@ if __name__ == "__main__":
         train_data,
         train_labels,
     )
+    
+    try:
+    import numpy as np
+    print("NumPy is properly installed and working.")
+except ImportError:
+    print("NumPy is not installed. Please install it using 'pip install numpy'.")
+
 
     print(f"预测的标签: {predicted_label}")