alexandre-huynh commited on
Commit
29ef7b0
1 Parent(s): a91e32c

Updated logs printing format

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -74,7 +74,8 @@ def format_text(text, to_remove):
74
  def generate_text(prompt):
75
  output = gpt2_lm.generate(prompt, max_length=150)
76
  formatted_output = format_text(output, prompt)
77
- print(f"DEBUG - GPT-2 Output : {output}")
 
78
  print(f"DEBUG - Formatted GPT-2 Output : {formatted_output}")
79
  return formatted_output
80
 
 
74
  def generate_text(prompt):
75
  output = gpt2_lm.generate(prompt, max_length=150)
76
  formatted_output = format_text(output, prompt)
77
+ print("=====================================")
78
+ print(f"DEBUG - GPT-2 Output : {output}\n")
79
  print(f"DEBUG - Formatted GPT-2 Output : {formatted_output}")
80
  return formatted_output
81