DarkNeuron-AI commited on
Commit
e4c7716
·
verified ·
1 Parent(s): 7344c33

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -60,8 +60,8 @@ import joblib
60
  import pandas as pd
61
 
62
  # Download and load the trained pipeline
63
- pipeline_path = hf_hub_download("HydraSense/waterpotabler-v1", "water_potability_pipeline.pkl")
64
- pipeline = joblib.load(pipeline_path)
65
 
66
  # Example water sample
67
  sample_data = {
@@ -79,7 +79,7 @@ sample_data = {
79
  sample_df = pd.DataFrame(sample_data)
80
 
81
  # Predict potability
82
- prediction = pipeline.predict(sample_df)
83
 
84
  print("Prediction:", "💧 Potable" if prediction[0] == 1 else "⚠️ Not Potable")
85
  ```
 
60
  import pandas as pd
61
 
62
  # Download and load the trained pipeline
63
+ pipeline_path = hf_hub_download("DarkNeuron-AI/darkneuron-hydrasense-v1", "water_potability_model.pkl")
64
+ model = joblib.load(pipeline_path)
65
 
66
  # Example water sample
67
  sample_data = {
 
79
  sample_df = pd.DataFrame(sample_data)
80
 
81
  # Predict potability
82
+ prediction = model.predict(sample_df)
83
 
84
  print("Prediction:", "💧 Potable" if prediction[0] == 1 else "⚠️ Not Potable")
85
  ```