Update app.py

app.py

@@ -5,11 +5,16 @@ import torch
 import json
 import pickle
 from pytorch_tabular import TabularModel
-
+from sklearn.preprocessing import LabelEncoder
 
 # Load model
 model = TabularModel.load_model("FTTransformerModel")
 
+#Encoder
+dm = model.datamodule
+if hasattr(dm, "label_encoder") and isinstance(dm.label_encoder, LabelEncoder):
+    dm.label_encoder = [dm.label_encoder]
+
 # Load threshold
 with open("FTTransformerModel/threshold.json", "r") as f:
     threshold = json.load(f)["threshold"]