From 4e9beec97c1a57e9708b3889b4402ca0b02029f5 Mon Sep 17 00:00:00 2001
From: Ethan Ng <ethanenguyen@hotmail.com>
Date: Tue, 19 Dec 2023 09:30:13 -0800
Subject: [PATCH] Update README.md

---
 README.md | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index c0828c1..30ab971 100644
--- a/README.md
+++ b/README.md
@@ -35,21 +35,24 @@ pip install sdr_hazards_classification
 :airplane: Follow the code snippet below to test and call the prediction method from the Depressurization model
 
 ```
-from sdr_hazards_classification import sdr_api
+from sdr_hazards_classification.sdr_api import SdrInferenceAPI, CORROSION_LIMIT, DEGRADED_CONTROLLABILITY
 import pandas as pd
 
-my_model = sdr_api.SdrInferenceAPI()
+depressurization_model = sdr_api.SdrInferenceAPI()
 
 #test the prediction method
-my_model.test_sdr_depressurization_predictions()
+depressurization_model.test_sdr_depressurization_predictions()
 
 event_text = "Lost cabin pressurization at flight level 30000, cabin altitude warning horn sounded at 10000 feet. Unabel to control cabin pressure with outflow valve closed"
-pred, probs = my_model.get_predictions([event_text])
+pred, probs = depressurization_model.get_predictions([event_text])
+
+degraded_controllability_model = SdrInferenceAPI(DEGRADED_CONTROLLABILITY)
+degraded_controllability_model.test_sdr_degraded_controllability()
 
 df = pd.read_csv('./src/sdr_classifier/data/SDR_Example.csv')
 records = df["Text"]
 #pass in a record list for prediction
-pred, probs = my_model.get_predictions(records)
+pred, probs = depressurization_model.get_predictions(records)
 
 df['Prediction'] = pred
 df['Prob'] = probs