Life-Hack-Generator/textGenerator.py at main · njfredri/Life-Hack-Generator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import gpt_2_simple as gpt2
import os
import requests
import sys
import threading
class textGenerator:
	def __init__(self):
		self.model_name = "noanimeChat"
		self.sess = gpt2.start_tf_sess()
		gpt2.load_gpt2(sess=self.sess,model_dir='checkpoint',model_name=self.model_name)

	def generateText(self):
		print("clearing text")
		tempfile = open('temp.txt','w')
		tempfile.truncate(0)
		tempfile.close()
		# model_name = "shrek"
		# sess = gpt2.start_tf_sess()
		# gpt2.load_gpt2(sess=sess,model_dir='checkpoint',model_name=model_name)
		# print("loaded. Now creating text")
		gpt2.generate(sess=self.sess, destination_path='temp.txt')
		print(sess)
		print("Ended")

	def generateText(self):
		print("clearing text")
		tempfile = open('temp.txt','w')
		tempfile.truncate(0)
		tempfile.close()
		print(self.sess)
		# model_name = "shrek"
		# sess = gpt2.start_tf_sess()
		# gpt2.load_gpt2(sess=sess,model_dir='checkpoint',model_name=model_name)
		# print("loaded. Now creating text")
		gpt2.generate(sess=self.sess, destination_path='temp.txt')
		print("Ended")

# if __name__== '__main__':
# 	x = textGenerator()
# 	x.generateTextInAThread()
# 	print("waiting on thread")