def top_k(stub, input_sentence, model_name, length): input_msg = gpt_2_server_pb2.GenMsg(input_seed_sentence=input_sentence, input_model_name=model_name, input_length=length, output_generated_text="none") output_msg = stub.Top_k(input_msg).output_generated_text return output_msg
def improvise(stub, model_name, length): input_msg = gpt_2_server_pb2.GenMsg(input_seed_sentence="None", input_model_name=model_name, input_length=length, output_generated_text="none") output_msg = stub.Improvise(input_msg).output_generated_text return output_msg
def Improvise(self, request, context): print(request.input_model_name) generated_text = self.network.generate( model_name=request.input_model_name, opperation=Opperation.IMPROVISE, length=request.input_length) return gpt_2_server_pb2.GenMsg( input_seed_sentence="None", input_model_name=request.input_model_name, output_generated_text=generated_text)
def Top_k(self, request, context): print(request.input_seed_sentence) generated_text = self.network.generate( seed_sentence=request.input_seed_sentence, model_name=request.input_model_name, opperation=Opperation.SINGLE_WORD, length=request.input_length) return gpt_2_server_pb2.GenMsg( input_seed_sentence=request.input_seed_sentence, input_model_name=request.input_model_name, output_generated_text=generated_text)