Update src/streamlit_app.py

src/streamlit_app.py

@@ -17,6 +17,23 @@ from orb_models.forcefield.calculator import ORBCalculator
 #from mattersim.forcefield import MatterSimCalculator
 import pandas as pd
 import yaml # Added for FairChem reference energies
+import subprocess
+import sys
+import pkg_resources
+
+# Define version threshold
+required_version = "2.0.0"
+
+try:
+    installed_version = pkg_resources.get_distribution("numpy").version
+    if pkg_resources.parse_version(installed_version) >= pkg_resources.parse_version(required_version):
+        print(f"numpy version {installed_version} >= {required_version}. Installing numpy<2.0.0...")
+        subprocess.check_call([sys.executable, "-m", "pip", "install", "numpy<2.0.0"])
+    else:
+        print(f"numpy version {installed_version} is already < {required_version}. No action needed.")
+except pkg_resources.DistributionNotFound:
+    print("numpy is not installed. Installing numpy<2.0.0...")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "numpy<2.0.0"])
 
 from huggingface_hub import login