up

app.py

@@ -4,12 +4,13 @@ import glob
 import shutil
 import os
 import tqdm
+from huggingface_hub import snapshot_download
 import tempfile
 import re
 import pdfminer
 
-print("pdfminer", print(pdfminer.__version__))
-print("pandoc", print(pypandoc.__version__))
+print("pdfminer", pdfminer.__version__)
+print("pandoc", pypandoc.__version__)
 
 #from docx import Document
 #document = Document()
@@ -98,11 +99,12 @@ def main(filename, codewords_mapping):
                 f.write(out)
 
 def convert(*keywords):
+    # cached_folder = snapshot_download("claudiag/atlas", token=os.environ.get("HF_TOKEN"))
     codewords_mapping = {k: v for k,v in zip(CATEGORIES, keywords)}
 
     num_files = 0
 
-    return "_".join(codewords_mapping.keys())
+    return "_".join(codewords_mapping.values())
 
     for folder in tqdm.tqdm(glob.glob("./*")):
         shutil.rmtree(RESULTS_FOLDER, ignore_errors=True)