From fd838ac5abce36259fc7cb78ebcce0cf77aa55a2 Mon Sep 17 00:00:00 2001 From: tharoosha Date: Sat, 28 Oct 2023 15:02:19 +0530 Subject: [PATCH] docker path changes --- backend/ml_models/emotion_detection/emotionScript.py | 6 ++++-- .../spotifyRecommendExecution.py | 9 ++++++--- .../spotify_recommendation/spotifyRecommendScript.py | 10 +++++++--- 3 files changed, 17 insertions(+), 8 deletions(-) diff --git a/backend/ml_models/emotion_detection/emotionScript.py b/backend/ml_models/emotion_detection/emotionScript.py index def7d89..ae8220f 100644 --- a/backend/ml_models/emotion_detection/emotionScript.py +++ b/backend/ml_models/emotion_detection/emotionScript.py @@ -126,10 +126,12 @@ def initialize(): ''' Load the model and tokenizer from the saved files ''' - with open('../backend/ml_models/emotion_detection/tokenizer.pkl', 'rb') as f: + with open('/usr/src/app/ml_models/emotion_detection/tokenizer.pkl', 'rb') as f: + # with open('../backend/ml_models/emotion_detection/tokenizer.pkl', 'rb') as f: tokenizer = pickle.load(f) - model = tf.keras.models.load_model('../backend/ml_models/emotion_detection/emotion_model') + model = tf.keras.models.load_model('/usr/src/app/ml_models/emotion_detection/emotion_model') + # model = tf.keras.models.load_model('../backend/ml_models/emotion_detection/emotion_model') ################################################################################################################################################################################# diff --git a/backend/ml_models/spotify_recommendation/spotifyRecommendExecution.py b/backend/ml_models/spotify_recommendation/spotifyRecommendExecution.py index ccac235..462adad 100644 --- a/backend/ml_models/spotify_recommendation/spotifyRecommendExecution.py +++ b/backend/ml_models/spotify_recommendation/spotifyRecommendExecution.py @@ -22,12 +22,15 @@ def script_run(input_mood): try: - file_path = '../backend/ml_models/spotify_recommendation/tokenizer.pkl' + file_path = '/usr/src/app/ml_models/spotify_recommendation/tokenizer.pkl' + # file_path = '../backend/ml_models/spotify_recommendation/tokenizer.pkl' if os.path.exists(file_path): - with open('../backend/ml_models/spotify_recommendation/tokenizer.pkl', 'rb') as f: + with open('/usr/src/app/ml_models/spotify_recommendation/tokenizer.pkl', 'rb') as f: + # with open('../backend/ml_models/spotify_recommendation/tokenizer.pkl', 'rb') as f: scaler = pickle.load(f) - model = tf.keras.models.load_model('../backend/ml_models/spotify_recommendation/spotify_model') + model = tf.keras.models.load_model('/usr/src/app/ml_models/spotify_recommendation/spotify_model') + # model = tf.keras.models.load_model('../backend/ml_models/spotify_recommendation/spotify_model') result = spotifyRecommendScript.getRecommendation(input_mood, model, scaler) # result = json.dumps(result) diff --git a/backend/ml_models/spotify_recommendation/spotifyRecommendScript.py b/backend/ml_models/spotify_recommendation/spotifyRecommendScript.py index bfbfde2..4515d2b 100644 --- a/backend/ml_models/spotify_recommendation/spotifyRecommendScript.py +++ b/backend/ml_models/spotify_recommendation/spotifyRecommendScript.py @@ -90,7 +90,9 @@ def initialize(): ####################################################################################################################################### '''Train model from scratch and save it along with the scaler''' - df=pd.read_csv("backend/ml_models/spotify_recommendation/dataset.csv") + df=pd.read_csv("/usr/src/app/ml_models/spotify_recommendation/dataset.csv") + # df=pd.read_csv("backend/ml_models/spotify_recommendation/dataset.csv") + df['Mood'] = df['Mood'].apply(class_to_index) @@ -127,10 +129,12 @@ def initialize(): print(f"Test loss: {loss:.4f}, Test accuracy: {accuracy:.4f}") # Save the tokenizer - with open('backend/ml_models/spotify_recommendation/tokenizer.pkl', 'wb') as f: + with open('/usr/src/app/ml_models/spotify_recommendation/tokenizer.pkl', 'wb') as f: + # with open('backend/ml_models/spotify_recommendation/tokenizer.pkl', 'wb') as f: pickle.dump(scaler, f) - model.save('backend/ml_models/spotify_recommendation/spotify_model') + model.save('/usr/src/app/ml_models/spotify_recommendation/spotify_model') + # model.save('backend/ml_models/spotify_recommendation/spotify_model') #######################################################################################################################################