Update README.md

README.md

@@ -28,3 +28,31 @@ It is saved as a pickle file: `model.pkl` and includes all custom layers (e.g.,
 
 ```bash
 pip install torch pandas
+
+
+import torch
+
+# Load the model from pickle
+with open("model.pkl", "rb") as f:
+    model = torch.load(f)
+
+model.eval()
+
+
+# Example dummy input
+x_num = torch.rand((1, 10))          # Replace 10 with your actual num_features
+x_cat = torch.randint(0, 5, (1, 3))  # Replace with your actual number of categories
+
+# Predict
+churn_prob, predicted_tenure, predicted_ltv = model(x_num, x_cat)
+
+print("Churn probability:", churn_prob.item())
+print("Predicted tenure:", predicted_tenure.item())
+print("Predicted LTV:", predicted_ltv.item())
+
+
+(
+    churn_prob: FloatTensor of shape (B, 1),
+    predicted_tenure: FloatTensor of shape (B, 1),
+    predicted_ltv: FloatTensor of shape (B, 1)
+)