diff --git a/AIDungeon_2.ipynb b/AIDungeon_2.ipynb
index e828288..4a53e1d 100644
--- a/AIDungeon_2.ipynb
+++ b/AIDungeon_2.ipynb
@@ -20,12 +20,23 @@
"colab_type": "text"
},
"source": [
- "\n",
+ "\n",
"\n",
- "Sponsored by the [BYU PCCL Lab](https://).\n",
+ "Sponsored by the BYU PCCL Lab.\n",
"\n",
- "# AI Dungeon 2 is currently down due to high download costs. \n",
- "# on twitter for updates on when it will be available again.\n",
+ "> AI Dungeon 2 is a completely AI generated text adventure built with OpenAI's largest GPT-2 model. It's a first of it's kind game that allows you to enter and will react to any action you can imagine.\n",
+ "\n",
+ "# Main mirrors of AI Dungeon 2 are currently down due to high download costs.\n",
+ "We are using bittorrent as a temporary solution to host game files and keep this game alive. It's not fast, but it's the best we've got right now.\n\n",
+ "If you want to help, best thing you can do is to **[download this torrent file with game files](https://github.com/nickwalton/AIDungeon/files/3935881/model_v5.torrent.zip)** and **seed it** indefinitely to the best of your ability. This will help new players download this game faster, and discover the vast worlds of AIDungeon2!\n\n",
+ "- on twitter for updates on when it will be available again.\n",
+ "- **[Support AI Dungeon 2 on Patreon to help get it back up!](https://www.patreon.com/posts/update-32193930)**\n",
+ "\n",
+ "## How to play\n",
+ "1. Click \"Tools\"-> \"Settings...\" -> \"Theme\" -> \"Dark\" (optional but recommended)\n",
+ "2. Click \"Runtime\" -> \"Run all\"\n",
+ "3. Wait until all files are downloaded (only has to be one once, and it will take some time)\n",
+ "4. It will then take a couple minutes to boot up as the model is downloaded loaded onto the GPU. \n",
"\n",
"## About\n",
"* While you wait you can [read adventures others have had](https://aidungeon.io/)\n",
@@ -42,7 +53,7 @@
"colab": {}
},
"source": [
- "!git clone --depth 1 --branch master https://github.com/nickwalton/AIDungeon/\n",
+ "!git clone --depth 1 --branch master https://github.com/AIDungeon/AIDungeon/\n",
"%cd AIDungeon\n",
"!./install.sh\n",
"from IPython.display import clear_output \n",
@@ -68,4 +79,4 @@
"outputs": []
}
]
-}
\ No newline at end of file
+}
diff --git a/README.md b/README.md
index 9856712..e2eb78f 100644
--- a/README.md
+++ b/README.md
@@ -7,9 +7,9 @@ Read more about AIDungeon2 and how it was built [here](https://pcc.cs.byu.edu/20
Play the game in Colab [here](http://www.aidungeon.io).
-To play the game locally you need a beefy GPU ~12 GB and CUDA installed.
+To play the game locally, it is recommended that you have an nVidia GPU with 12 GB or more of memory, and CUDA installed. If you do not have such a GPU, each turn can take a couple of minutes or more for the game to compose its response. To install and play locally:
```
-git clone https://github.com/nickwalton/AIDungeon/
+git clone https://github.com/AIDungeon/AIDungeon/
cd AIDungeon
./install.sh
python play.py
@@ -28,6 +28,7 @@ Resources:
* **Website**: [aidungeon.io](http://www.aidungeon.io/)
* **Email**: aidungeon.io@gmail.com
* **Twitter**: [@nickwalton00](https://twitter.com/nickwalton00), [@benjbay](https://twitter.com/benjbay)
+* **Reddit**: [r/AIDungeon](https://www.reddit.com/r/AIDungeon/)
Contributing
diff --git a/install.sh b/install.sh
index 827e680..d5c02d4 100755
--- a/install.sh
+++ b/install.sh
@@ -24,14 +24,13 @@ else
echo -e "\n\n==========================================="
echo "We are now starting to download the model."
echo "It will take a while to get up to speed."
- echo "After download completes, we will seed this for 1 minute to ensure high availability."
echo "DHT errors are normal."
echo -e "===========================================\n"
aria2c \
--max-connection-per-server 16 \
--split 64 \
--bt-max-peers 500 \
- --seed-time=1 \
+ --seed-time=0 \
--summary-interval=15 \
--disable-ipv6 \
"${MODEL_TORRENT_BASENAME%.*}"
diff --git a/play.py b/play.py
index dfef418..211efcd 100644
--- a/play.py
+++ b/play.py
@@ -75,7 +75,7 @@ def play_aidungeon_2():
story_manager = UnconstrainedStoryManager(generator)
print("\n")
- with open('opening.txt', 'r') as file:
+ with open('opening.txt', 'r', encoding='utf-8') as file:
starter = file.read()
print(starter)
diff --git a/requirements.txt b/requirements.txt
index 477ca9f..670bb7a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,6 @@
google-cloud-storage
numpy
regex
-profanityfilter
\ No newline at end of file
+profanityfilter
+pyyaml
+tensorflow==1.15