Spaces:
Runtime error
Runtime error
paths updated 2
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ class ImprovedSongRecommender(nn.Module):
|
|
27 |
return x
|
28 |
|
29 |
# Load the trained model
|
30 |
-
model_path = "models
|
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
|
39 |
-
scaler_path = "data
|
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)
|