Update inference.py

inference.py

@@ -8,6 +8,12 @@ import glob
 from sklearn.multioutput import MultiOutputRegressor
 from tabpfn_extensions.post_hoc_ensembles.sklearn_interface import AutoTabPFNRegressor
 
+
+print(os.environ.get("OMP_NUM_THREADS"))
+os.environ["OMP_NUM_THREADS"] = os.cpu_count()
+
+
+
 import torch
 
 # Give torch.classes a benign __path__ so Streamlit won't trigger __getattr__.