diff --git a/app/app.py b/app/app.py
index ab9e45f..4a8b74e 100644
--- a/app/app.py
+++ b/app/app.py
@@ -9,7 +9,7 @@
 
 app = Flask(__name__)
 
-Model, config_file = load_model("./src/Baseline_model/state_dict.py", "./src/Baseline_model/config_model.json") # argparse
+Model, config_file = load_model("../src/Baseline_model/state_dict.pt", "../src/Baseline_model/config_model.json") # argparse
 tokenizer = load_tokenizer()
 
 @app.route('/predict', methods = ['POST'])