diff --git a/movies.py b/movies.py
index 088e27d89c724d88bbf608328d729d019bdda621..dc8f8e564b14b208ec9a12447cc522753741ff71 100644
--- a/movies.py
+++ b/movies.py
@@ -350,15 +350,6 @@ def log_preduction_model(model):
                            pip_requirements="requirements.txt")
 
 
-def build_prediction_model(model, movies):
-    """Build an index to get movie suggestions"""
-    index = tfrs.layers.factorized_top_k.BruteForce(model.query_model)
-    index.index_from_dataset(tf.data.Dataset.zip(
-        (movies.batch(100), movies.batch(100).map(model.cand_model))
-    ))
-    return index
-
-
 def train(args):
     """Load the dataset, train and evaluate the model"""
     mlflow.set_experiment(args.mlflow_experiment)