From 594ba6411db311976652774b7764f1b1c71ff6e1 Mon Sep 17 00:00:00 2001 From: Javclaude Date: Sun, 31 May 2020 18:31:22 +0200 Subject: [PATCH] Fix path --- app/app.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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'])