Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 import pandas as pd
 from pytorch_tabular import TabularModel
-from pytorch_tabular.config import DataConfig, ModelConfig, TrainerConfig
+from pytorch_tabular.config import DataConfig, TrainerConfig
+from pytorch_tabular.models import CategoryEmbeddingModelConfig
 
 # Sample Data
 data = {
@@ -17,7 +18,7 @@ data_config = DataConfig(
     target=["target"],
     continuous_cols=["feature1", "feature2", "feature3"]
 )
-model_config = ModelConfig(task="classification", num_classes=2)
+model_config = CategoryEmbeddingModelConfig(task="classification")  # No `num_classes`
 trainer_config = TrainerConfig(max_epochs=10)
 
 # Initialize and train model