Update src/predict.py

src/predict.py

@@ -20,7 +20,7 @@ def predict_single(model, sample_dict):
 
 if __name__ == "__main__":
     # Example usage with Kidney Disease model
-    model_path_kidney = "/content/models/model_kidney.joblib" # Path to the saved kidney model
+    model_path_kidney = "/content/models/kidney_model.pkl" # Path to the saved kidney model
     model_kidney = load_model(model_path_kidney)
 
     # Sample data for Kidney Disease prediction
@@ -92,7 +92,7 @@ if __name__ == "__main__":
     print("Kidney Disease Prediction:", res_kidney)
 
     # Example usage with Liver Patient model
-    model_path_liver = "/content/models/model_liver.joblib"
+    model_path_liver = "/content/models/liver_model.pkl"
     model_liver = load_model(model_path_liver)
     # Sample data for Liver Patient prediction
     # Include all columns from the merged dataset, fill with relevant values or None/np.nan