peter szemraj commited on
Commit
32c181d
1 Parent(s): 9ba1d56

update placeholder

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -39,7 +39,9 @@ def generate_text(
39
  str: the generated text
40
  """
41
  global generator
42
- logging.info(f"Generating text from prompt: {prompt}")
 
 
43
  st = time.perf_counter()
44
 
45
  input_tokens = generator.tokenizer(prompt)
@@ -97,7 +99,7 @@ def get_parser():
97
  default_prompt = """
98
  Hello,
99
 
100
- Got this in my personal email, might be very useful as a "real" platform for a hackathon. Let"""
101
 
102
  if __name__ == "__main__":
103
  logging.info("\n\n\nStarting new instance of app.py")
 
39
  str: the generated text
40
  """
41
  global generator
42
+ if verbose:
43
+ logging.info(f"Generating text from prompt:\n\n{prompt}")
44
+ logging.info(f"params:\tmax_length={gen_length}, num_beams={num_beams}, no_repeat_ngram_size={no_repeat_ngram_size}, length_penalty={length_penalty}, repetition_penalty={repetition_penalty}, abs_max_length={abs_max_length}")
45
  st = time.perf_counter()
46
 
47
  input_tokens = generator.tokenizer(prompt)
 
99
  default_prompt = """
100
  Hello,
101
 
102
+ Following up on last's bubblegum shipment, I"""
103
 
104
  if __name__ == "__main__":
105
  logging.info("\n\n\nStarting new instance of app.py")