diff options
-rw-r--r-- | bot.py | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -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) |