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()