Spaces:

davanstrien
/

dataset-language-prediction-api-literstar

Sleeping

davanstrien HF Staff commited on Jan 24, 2024

Commit

72db6d0

1 Parent(s): edb0177

Update fasttext model download parameters

Files changed (1) hide show

main.py CHANGED Viewed

@@ -144,7 +144,10 @@ FASTTEXT_PREFIX_LENGTH = 9  # fasttext labels are formatted like "__label__eng_L
 model = fasttext.load_model(
     hf_hub_download(
-        "facebook/fasttext-language-identification", "model.bin", cache_dir="."
     )
 )

 model = fasttext.load_model(
     hf_hub_download(
+        "facebook/fasttext-language-identification",
+        "model.bin",
+        cache_dir=".",
+        download_dir=".",
     )
 )