diff --git a/test.py b/test.py index ce6eef6..0a31474 100644 --- a/test.py +++ b/test.py @@ -5,9 +5,9 @@ from synthesis_action_retriever.build_graph import GraphBuilder from synthesis_action_retriever.utils import make_spacy_tokens -dir_path = "/Users/kevcruse96/Desktop/D2S2/Saved_Models/sar_models" -w2v_model = "w2v_embeddings_v2_words_420K" -ext_model = "Bi-RNN_cl7_ed100_TF_20211018-122820" +dir_path = "path-to-models" +w2v_model = "path-to-w2v_model" +ext_model = "path-to-ext_model" sar = SynthActionRetriever( embedding_model=os.path.join(dir_path, w2v_model),