Make sure post is at least 100 characters

1 files changed, 5 insertions, 2 deletions

diff --git a/bot.py b/bot.py
index 6c37fa6..ea86b88 100644
--- a/bot.py
+++ b/bot.py
@@ -20,7 +20,7 @@ model = AutoModelForCausalLM.from_pretrained(args.model)
 # Run the input through the model
 inputs = tokenizer.encode(args.input, return_tensors="pt")
 output = tokenizer.decode(model.generate(
-    inputs, do_sample=True, max_length=25, top_p=0.9, temperature=0.8)[0])
+    inputs, do_sample=True, max_length=75, top_p=0.9)[0])
 print(output)
 
 
@@ -29,4 +29,7 @@ mastodon = Mastodon(
     access_token=args.token,
     api_base_url=args.instance
 )
-mastodon.status_post(output)
+post = output.split('\n')[0]
+if len(post) < 100:
+    post = output.split('\n')[0] + '\n' + output.split('\n')[1]
+mastodon.status_post(post)