Link Search Menu Expand Document

Text Generation Basic Usage

generate_text()

The method generate_text() contains 2 arguments:

  1. text (string): The text prompt for the model – it will try to continue the text
  2. args (GENSettings): See this webpage for more information

Returns: An object with a single field called “text”

Example 1.1:


from happytransformer import HappyGeneration, GENSettings
#--------------------------------------#
    happy_gen = HappyGeneration()  # default uses gpt2
    args = GENSettings(max_length=15)
    result = happy_gen.generate_text("artificial intelligence is ", args=args)    
    print(result)  # GenerationResult(text='\xa0a new field of research that has been gaining momentum in recent years.')
    print(result.text)  #  a new field of research that has been gaining momentum in recent years.