Fix path

HessTaha · May 31, 2020 · 594ba64 · 594ba64
1 parent c0c8b81
commit 594ba64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'])