Joash2024 commited on
Commit
3533ee2
·
1 Parent(s): e1d942e

paths updated 2

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 = "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 = "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)
 
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)