diff --git a/pred_from_model.py b/pred_from_model.py
index 7d2acb7..71f4c9f 100644
--- a/pred_from_model.py
+++ b/pred_from_model.py
@@ -2,7 +2,7 @@ import json
 import mlflow as mlf
 import pandas as pd
 
-model = mlf.keras.load_model("country_vaccination")
+model = mlf.keras.load_model("movies_imdb")
 
 
 with open('movies_imdb/input_example.json', 'r') as datafile: