Merge pull request #9 from SkyDocs/beta

Beta

README.md

@@ -32,13 +32,13 @@ Install the requirements by running:
 
 Change the path to *noise* dataset in `predict.py` file to your own path to *noise* folder.
 
-After installing the requirements and doing the necessary changes. 
+After installing the requirements and doing the necessary changes.
 
 #### CLI
 
 Run:
 
-`python cli/voice_assistant.py` or `python3 cli/voice_assistant.py` <br>
+`python voice_assistant_cli.py` or `python3 voice_assistant_cli.py` <br>
 
 And here you go. <br>
 

cli/predict.py

@@ -15,7 +15,9 @@
 from IPython.display import display, Audio
 import sys
 sys.path.insert(0, "../")
-from users import user_0, user_1, user_2, user_new
+print(os.getcwd())
+print(os.listdir())
+from users import user_0, user_1, user_2
 
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
@@ -244,12 +246,12 @@ def predict(path, labels):
 				print("\033[31m[*]\033[0m Welcome user 2")
 				user_2.predict()
 			else:
-				print("\033[31m[*]\033[0m Welcome new user")
-				user_new.predict()
+				print("\033[31m[*]\033[0m Sorry User not recognised")
+				pass
 
 
 """ Predict """
 path = ["predict.wav"]
 labels = ["unknown"]
 model = tf.keras.models.load_model('../model.h5')
-predict(path, labels)
+predict(path, labels)