Update inference.py

inference.py

@@ -1,6 +1,6 @@
 import pandas as pd
 import numpy as np
-import torch
+# import torch
 import joblib
 import argparse
 import os
@@ -8,6 +8,22 @@ import glob
 from sklearn.multioutput import MultiOutputRegressor
 from tabpfn_extensions.post_hoc_ensembles.sklearn_interface import AutoTabPFNRegressor
 
+import torch
+
+# Give torch.classes a benign __path__ so Streamlit won't trigger __getattr__.
+try:
+    setattr(torch.classes, "__path__", [])
+except Exception:
+    # Fallback wrapper if direct setattr isn't allowed in your build
+    class _TorchClassesWrapper:
+        def __init__(self, obj):
+            self._obj = obj
+            self.__path__ = []
+        def __getattr__(self, name):
+            return getattr(self._obj, name)
+    torch.classes = _TorchClassesWrapper(torch.classes)
+
+
 class TabPFNEnsemblePredictor:
 	"""
 	A class to load an ensemble of TabPFN models and generate averaged predictions.