Update ModelSimilarity.py

ModelSimilarity.py

@@ -145,7 +145,7 @@ class MovieSimilarityModel:
         top_5_genres = self.dataset['genres'].iloc[top_5_indices].values
         
         # Return the top 5 movies with their similarity scores
-        return pd.DataFrame(list(zip(top_5_movies, top_5_scores,top_5_plots,top_5_genres)), columns=['Movie', 'Similarity Score'])
+        return pd.DataFrame(list(zip(top_5_movies, top_5_scores,top_5_plots,top_5_genres)), columns=['Movie', 'Similarity Score','Plot','Genres'])
 
     def load_tensor_list(self):
         # Load precomputed embeddings and map them to CUDA