Update README.md
Browse files
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("
|
| 64 |
-
|
| 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 =
|
| 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 |
```
|