Add support for more metadata

jcntrl · Dec 13, 2023 · f159319 · f159319
1 parent 563dfd1
commit f159319
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/flashrank/Ranker.py b/flashrank/Ranker.py
@@ -7,8 +7,7 @@
 import zipfile
 import requests
 from tqdm import tqdm
-# from flashrank.Config import default_model, default_cache_dir, model_url, model_file_map
-from Config import default_model, default_cache_dir, model_url, model_file_map
+from flashrank.Config import default_model, default_cache_dir, model_url, model_file_map
 import collections
 
 class RerankRequest: