diff --git a/server/attention_rnn.mag b/server/attention_rnn.mag
new file mode 100644
index 0000000..b10019c
Binary files /dev/null and b/server/attention_rnn.mag differ
diff --git a/server/predict.py b/server/predict.py
index 640fe86..27d0258 100644
--- a/server/predict.py
+++ b/server/predict.py
@@ -29,7 +29,7 @@ import time
 import tempfile
 import pretty_midi
 
-BUNDLE_NAME = 'basic_rnn'
+BUNDLE_NAME = 'attention_rnn'
 
 config = magenta.models.melody_rnn.melody_rnn_model.default_configs[BUNDLE_NAME]
 bundle_file = magenta.music.read_bundle_file(os.path.abspath(BUNDLE_NAME+'.mag'))