Spaces:

davanstrien
/

dataset-language-prediction-api-literstar

Sleeping

davanstrien HF Staff commited on Jan 24, 2024

Commit

edb0177

1 Parent(s): cb7fe33

Update fasttext model download with cache directory

Files changed (1) hide show

main.py CHANGED Viewed

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

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