Update app.py

app.py

@@ -13,21 +13,33 @@ data = {
 }
 df = pd.DataFrame(data)
 
-# Configure pytorch_tabular
+# Ensure all configurations are set correctly
 data_config = DataConfig(
     target=["target"],
-    continuous_cols=["feature1", "feature2", "feature3"]
+    continuous_cols=["feature1", "feature2", "feature3"],
+    task="classification"
 )
-model_config = CategoryEmbeddingModelConfig(task="classification")  # No `num_classes`
-trainer_config = TrainerConfig(max_epochs=10)
-
-# Initialize and train model
-tabular_model = TabularModel(
-    data_config=data_config,
-    model_config=model_config,
-    trainer_config=trainer_config
+
+model_config = CategoryEmbeddingModelConfig(
+    task="classification",
+    layers="64-64",  # Example hidden layer sizes
+    learning_rate=1e-3
+)
+
+trainer_config = TrainerConfig(
+    max_epochs=10
 )
-tabular_model.fit(df)
+
+# Initialize and train the model
+try:
+    tabular_model = TabularModel(
+        data_config=data_config,
+        model_config=model_config,
+        trainer_config=trainer_config
+    )
+    tabular_model.fit(df)
+except ValueError as e:
+    print(f"Error initializing TabularModel: {e}")
 
 # Define Inference Function
 def classify(feature1, feature2, feature3):
@@ -55,4 +67,3 @@ iface = gr.Interface(
 # Launch with additional server settings for Hugging Face Spaces
 print("Launching Gradio Interface...")
 iface.launch(server_name="0.0.0.0", server_port=7860, share=True)
-