Add instance flag for bot

1 files changed, 3 insertions, 2 deletions

diff --git a/bot.py b/bot.py
index 6dda1fd..6c37fa6 100644
--- a/bot.py
+++ b/bot.py
@@ -5,8 +5,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
 parser = ArgumentParser()
+parser.add_argument('-i', '--instance', help='Mastodon instance hosting the bot')
 parser.add_argument('-t', '--token', help='Mastodon application access token')
-parser.add_argument('-i', '--input', help='initial input text for prediction')
+parser.add_argument('-n', '--input', help='initial input text')
 parser.add_argument('-m', '--model', default='model',
                     help='path to load saved model')
 args = parser.parse_args()
@@ -26,6 +27,6 @@ print(output)
 # Post it to Mastodon
 mastodon = Mastodon(
     access_token=args.token,
-    api_base_url='https://social.exozy.me/'
+    api_base_url=args.instance
 )
 mastodon.status_post(output)