Update README.md

README.md

@@ -31,17 +31,17 @@ import sys
 bert_gts_pretrained = hf_hub_download(repo_id="gauneg/bert-gts-absa-triple", filename="bert_gts_pretrained.py")
 post =  hf_hub_download(repo_id="gauneg/bert-gts-absa-triple", filename="post.py")
 
-sys.path.append(bert_gts_pretrained)
-sys.path.append(post)
+sys.path.append(bert_gts_pretrained.rsplit("/", 1)[0])
+sys.path.append(post.rsplit("/", 1)[0])
+
 
-from bert_gts_model import GTSBertBaseABSATriple
-from token_decode_module import DecodeAndEvaluate
 from bert_gts_pretrained import GTSBertBaseABSATriple
 from post import DecodeAndEvaluate
 
 
 from transformers import AutoTokenizer
 
+
 model_id = 'gauneg/bert-gts-absa-triple'
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = GTSBertBaseABSATriple.from_pretrained(model_id)