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)