From 5d7be2475ad64ae65c6ca13f8febf42ed44b95cb Mon Sep 17 00:00:00 2001
From: insert username here <montyd18@gmail.com>
Date: Sat, 22 Jan 2022 13:49:46 -0800
Subject: [PATCH] Add files via upload

---
 promptai.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 promptai.py

diff --git a/promptai.py b/promptai.py
new file mode 100644
index 0000000..f543e5d
--- /dev/null
+++ b/promptai.py
@@ -0,0 +1,12 @@
+from aitextgen import aitextgen
+import sys
+
+s = sys.argv[1:]
+
+prompt1 = "" 
+for ele in s: 
+    prompt1 += ele
+
+ai = aitextgen(model_folder="trained_model", tokenizer_file="aitextgen.tokenizer.json")
+
+ai.generate(n=1, prompt=prompt1, max_length=500)