Update app.py

app.py

@@ -26,15 +26,16 @@ import pickle
 from pytorch_tabular import TabularModel
 from sklearn.preprocessing import LabelEncoder
 from omegaconf import OmegaConf, DictConfig
+from types import SimpleNamespace
 
 # Load model
 model = TabularModel.load_model("FTTransformerModel")
 
-model.datamodule._inferred_config = model._inferred_config
-
 dm = model.datamodule
 if hasattr(dm, "label_encoder") and isinstance(dm.label_encoder, LabelEncoder):
     dm.label_encoder = [dm.label_encoder]
+
+dm._inferred_config = SimpleNamespace(output_cardinality=[2])
     
 # Load threshold
 with open("FTTransformerModel/threshold.json", "r") as f: