Joash2024 commited on
Commit
e1d942e
·
1 Parent(s): 41f988c

paths updated

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -27,7 +27,7 @@ class ImprovedSongRecommender(nn.Module):
27
  return x
28
 
29
  # Load the trained model
30
- model_path = "C:/Users/joash/Desktop/Neurobytes_final_project/Neurobytes_Music_Recommender/models/improved_model.pth"
31
  num_unique_titles = 4855
32
 
33
  model = ImprovedSongRecommender(input_size=2, num_titles=num_unique_titles)
@@ -35,8 +35,8 @@ model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
35
  model.eval()
36
 
37
  # Load the label encoders and scaler
38
- label_encoders_path = "C:/Users/joash/Desktop/Neurobytes_final_project/Neurobytes_Music_Recommender/data/new_label_encoders.joblib"
39
- scaler_path = "C:/Users/joash/Desktop/Neurobytes_final_project/Neurobytes_Music_Recommender/data/new_scaler.joblib"
40
 
41
  label_encoders = load(label_encoders_path)
42
  scaler = load(scaler_path)
 
27
  return x
28
 
29
  # Load the trained model
30
+ model_path = "models\improved_model.pth"
31
  num_unique_titles = 4855
32
 
33
  model = ImprovedSongRecommender(input_size=2, num_titles=num_unique_titles)
 
35
  model.eval()
36
 
37
  # Load the label encoders and scaler
38
+ label_encoders_path = "data\new_label_encoders.joblib"
39
+ scaler_path = "data\new_scaler.joblib"
40
 
41
  label_encoders = load(label_encoders_path)
42
  scaler = load(scaler_path)