:art: format

app.py

@@ -41,7 +41,9 @@ def generate_text(
     global generator
     if verbose:
         logging.info(f"Generating text from prompt:\n\n{prompt}")
-        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}")
+        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}"
+        )
     st = time.perf_counter()
 
     input_tokens = generator.tokenizer(prompt)
@@ -128,9 +130,7 @@ if __name__ == "__main__":
         with gr.Column():
 
             gr.Markdown("## Generate Text")
-            gr.Markdown(
-                "Edit the prompt and parameters and press **Generate**!"
-            )
+            gr.Markdown("Edit the prompt and parameters and press **Generate**!")
             prompt_text = gr.Textbox(
                 lines=4,
                 label="Email Prompt",

utils.py

@@ -2,7 +2,8 @@
     utils.py - Utility functions for the project.
 """
 
-def postprocess(text:str):
+
+def postprocess(text: str):
     """
     postprocess - remove common values in scraped dataset
 
@@ -11,19 +12,19 @@ def postprocess(text:str):
     """
 
     replacements = {
-        "ENA":"<COMPANY>",
-        "Enron":"<COMPANY>",
-        "Enron Corporation":"<COMPANY>",
-        "Sony Pictures Entertainment":"<COMPANY>",
-        "Columbia Pictures":"<COMPANY>",
-        "Sony":"<COMPANY>",
-        "Columbia":"<COMPANY>",
-        "Hillary":"Jane",
-        "Clinton":"Smith",
-        "Amy":"Jane",
-        "Sara":"Jane",
-        "Harambe":"Jane",
-        "Pascal":"<PERSON>",
+        "ENA": "<COMPANY>",
+        "Enron": "<COMPANY>",
+        "Enron Corporation": "<COMPANY>",
+        "Sony Pictures Entertainment": "<COMPANY>",
+        "Columbia Pictures": "<COMPANY>",
+        "Sony": "<COMPANY>",
+        "Columbia": "<COMPANY>",
+        "Hillary": "Jane",
+        "Clinton": "Smith",
+        "Amy": "Jane",
+        "Sara": "Jane",
+        "Harambe": "Jane",
+        "Pascal": "<PERSON>",
     }
 
     # replace common values, also check lowercase
@@ -33,7 +34,8 @@ def postprocess(text:str):
 
     return text
 
+
 def clear(**kwargs):
-   """for use with buttons"""
+    """for use with buttons"""
 
-   return ""
+    return ""