From 54b451908a33873ccb494a5288770a204f74169d Mon Sep 17 00:00:00 2001
From: array-in-a-matrix <nova.neutrino@protonmail.com>
Date: Mon, 22 Aug 2022 00:27:32 -0400
Subject: [PATCH] optional parameter i think

---
 textgen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/textgen.py b/textgen.py
index f26bfbb..6a41a58 100644
--- a/textgen.py
+++ b/textgen.py
@@ -13,7 +13,7 @@ file_name = json_object['file']
 # ? generate message using trained model
 
 
-def generate_message(prompt):
+def generate_message(*prompt):
     ai = aitextgen(model_folder="trained_model",
                    tokenizer_file="aitextgen.tokenizer.json", prompt=prompt)
     ai.generate()