Update main.py

main.py

@@ -55,8 +55,8 @@ def generateText(model, minLen=100, size=5):
     return output
 if __name__ == "__main__":
     for model in models: # models = [(model, neg_model), (nsfw, neg_nsfw), (sfw, neg_sfw)]
-        text = generateText(model[0], k=k, minLen=300, size=5)
-        text_neg = generateText(model[1], k=k, minLen=300, size=5)
+        text = generateText(model[0], minLen=300, size=5)
+        text_neg = generateText(model[1], minLen=300, size=5)
 
         # print('\n'.join(text))
         for i in range(len(text)):