Update logic2.py

logic2.py

@@ -75,6 +75,17 @@ def load_model(data):
     model.eval()
     return model
 
+global id_map
+with open('id_map.pkl', 'rb') as file:
+    id_map = pickle.load(file)
+
+global m_id
+with open('m_id.pkl', 'rb') as file:
+    m_id = pickle.load(file)
+
+def get_movie(idx):
+    return id_map.loc[id_map['movieId'] == m_id[idx]].index
+
 def get_recommendation(model,data,user_id):
 
     total_movies = 9025
@@ -87,5 +98,6 @@ def get_recommendation(model,data,user_id):
     # we will only select movies for the user where the predicting rating is =5
     rec_movie_ids = (pred == 5).nonzero(as_tuple=True)
     top_ten_recs = [rec_movies for rec_movies in rec_movie_ids[0].tolist()[:10]] 
+    top_ten_recs = [get_movie(movie_idx) for movie_idx in top_ten_recs]
     return {'user': user_id, 'rec_movies': top_ten_recs}