diff --git a/download_model.sh b/download_model.sh deleted file mode 100755 index 1c82cf5..0000000 --- a/download_model.sh +++ /dev/null @@ -1 +0,0 @@ -gsutil -m cp -r gs://aidungeon2model/model_v1 ./aidungeon/generator/gpt2/models diff --git a/install.sh b/install.sh new file mode 100755 index 0000000..d074848 --- /dev/null +++ b/install.sh @@ -0,0 +1,10 @@ +MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v1 + +if [ -d "$MODEL_DIRECTORY" ]; then + echo "AIDungeon is already installed" + +else + echo "Installing AIDungeon" + gsutil -m cp -r gs://aidungeon2model/model_v1 ./aidungeon/generator/gpt2/models + pip install -r requirements.txt +fi