AhdCompnay commited on
Commit
8cb9a5d
1 Parent(s): 83dc256

Update kpe_ranker.py

Browse files
Files changed (1) hide show
  1. kpe_ranker.py +1 -1
kpe_ranker.py CHANGED
@@ -11,7 +11,7 @@ class KpeRanker:
11
  if os.path.isfile(model_path):
12
  TRAINED_MODEL_ADDR = model_path
13
  else:
14
- hf_hub_download(repo_id="ahdsoft/persian-keyphrase-extraction-model", filename="trained_model_10000.pt")
15
  TRAINED_MODEL_ADDR = model_path
16
 
17
  # TRAINED_MODEL_ADDR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'trained_model', 'trained_model_10000.pt')
 
11
  if os.path.isfile(model_path):
12
  TRAINED_MODEL_ADDR = model_path
13
  else:
14
+ hf_hub_download(repo_id="ahdsoft/persian-keyphrase-extraction-model", filename="trained_model_10000.pt", token="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")
15
  TRAINED_MODEL_ADDR = model_path
16
 
17
  # TRAINED_MODEL_ADDR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'trained_model', 'trained_model_10000.pt')