Update kpe_ranker.py

kpe_ranker.py

@@ -8,10 +8,10 @@ from huggingface_hub import hf_hub_download
 class KpeRanker:
     def __init__(self):
         model_path = "/root/.cache/huggingface/hub/models--ahdsoft--persian-keyphrase-extraction-model/trained_model_10000.pt"
-        if os.path.isfile(file_path):
+        if os.path.isfile(model_path):
             TRAINED_MODEL_ADDR = model_path
         else:
-            hf_hub_download(repo_id="lysandre/arxiv-nlp", filename="config.json")
+            hf_hub_download(repo_id="ahdsoft/persian-keyphrase-extraction-model", filename="trained_model_10000.pt")
             TRAINED_MODEL_ADDR = model_path
             
         # TRAINED_MODEL_ADDR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'trained_model', 'trained_model_10000.pt')