predict.py 文件源码

def predict():
    tf_config = tf.ConfigProto(gpu_options = tf.GPUOptions(visible_device_list = "0"))
    with tf.Session(config=tf_config) as sess:

        predictor = EasyPredictor(sess)
        sys.stdout.write("> ")
        sys.stdout.flush()
        line = sys.stdin.readline()
        while line:
            replies = predictor.predict(line)
            #for i, text in enumerate(replies):
                #print(i, text)
            print(replies[0])
            print("> ", end = "")
            sys.stdout.flush()
            line = sys.stdin.readline()