implemented main function

aa3c782d · W.D.R.P. Sandeepa · 448503c7 · aa3c782d
Commit aa3c782d authored Jan 03, 2022 by W.D.R.P. Sandeepa
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 0 deletions

backend/IT18218640/keyword_spotting_service.py backend/IT18218640/keyword_spotting_service.py +9 -0

No files found.
--- a/backend/IT18218640/keyword_spotting_service.py
+++ b/backend/IT18218640/keyword_spotting_service.py
@@ -55,3 +55,12 @@ def Keyword_Spotting_service():
        _Keyword_Spotting_Service.model = keras.models.load_model(MODEL_PATH)

    return _Keyword_Spotting_Service._instance
+
+if __name__ == "__main__":
+    kss = Keyword_Spotting_service()
+
+    keyword1 = kss.predict("test/0ab3b47d_nohash_0.wav")
+    keyword2 = kss.predict("test/blu7.wav")
+
+    print(f"Predicted Keywords: {keyword1}")
+    print(f"Predicted Keywords: {keyword2}")
\ No newline at end of file