fix embeddings script

wikispeedruns · Sep 3, 2024 · c95fa4d · c95fa4d
1 parent fa09000
commit c95fa4d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/get_embeddings.sh b/get_embeddings.sh
@@ -1,11 +1,11 @@
 #/bin/bash
 
 EMBEDDINGS_FILE="data/wiki2vec.txt.bz2"
-if [[ -f $EMBEDDINGS_FILE ]]; then
+if [[ ! -f $EMBEDDINGS_FILE ]]; then
     mkdir -p data
-    wget "http://wikipedia2vec.s3.amazonaws.com/models/en/2018-04-20/enwiki_20180420_100d.txt.bz2" -O $EMBEDDINGS_FILE.bz2
-    bunzip2 $EMBEDDINGS_FILE.bz2
+    wget "http://wikipedia2vec.s3.amazonaws.com/models/en/2018-04-20/enwiki_20180420_100d.txt.bz2" -O $EMBEDDINGS_FILE
+    bunzip2 $EMBEDDINGS_FILE
 else
     echo "\"$EMBEDDINGS_FILE\" already exists! Skipping..."
 
-fi
+fi