Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
-import joblib
 import numpy as np
+import pandas as pd
+import joblib
 from predict import extract_features
 import os
 import tempfile
@@ -9,6 +10,27 @@ import logging
 import traceback
 import sklearn
 
+# 版本检查
+import pkg_resources
+required_versions = {
+    'numpy': '1.23.5',
+    'scipy': '1.10.1',
+    'scikit-learn': '1.2.2'
+}
+
+def check_versions():
+    """检查包版本是否符合要求"""
+    for package, required_version in required_versions.items():
+        try:
+            installed_version = pkg_resources.get_distribution(package).version
+            st.write(f"{package} version: {installed_version} (required: {required_version})")
+            if installed_version != required_version:
+                st.warning(f"Warning: {package} version mismatch. Required: {required_version}, Installed: {installed_version}")
+        except pkg_resources.DistributionNotFound:
+            st.error(f"Error: {package} not found!")
+            return False
+    return True
+
 # Set up logging
 logging.basicConfig(
     level=logging.DEBUG,
@@ -23,6 +45,11 @@ st.set_page_config(
     layout="centered"
 )
 
+# 在加载模型之前检查版本
+if not check_versions():
+    st.error("Package version requirements not met. Please check the logs.")
+    st.stop()
+
 @st.cache_resource
 def load_model():
     """Load model from Hugging Face Hub"""
@@ -210,3 +237,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+