Update inference_gohan_cid.py

inference_gohan_cid.py

@@ -104,7 +104,7 @@ class GohanCIDInferenceEngine:
 		X_cat = torch.tensor([X_cat], dtype=torch.long)
 
 		# Numerical features (8 features to match training script)
-		num_cols = ['LAT', 'LONG', 'DELIVERY_NUM', 'MEDIAN_GENDER_RATIO', 'MODE_TOP_AGE_RANGE_1', 'MODE_TOP_AGE_RANGE_2', 'MODE_TOP_AGE_RANGE_3', 'TOTAL_VOLUME']
+		num_cols = ['LAT', 'LONG', 'DELIVERY_NUM', 'MEDIAN_GENDER_RATIO', 'TOTAL_VOLUME']
 		X_num = torch.tensor([[float(df.get(c, pd.Series([0])).iloc[0]) for c in num_cols]], dtype=torch.float32)
 		return X_cat, X_num