format

app.py

@@ -72,7 +72,8 @@ def generate_text(
     formatted_email = postprocess(response)
     return formatted_email, make_email_link(body=formatted_email)
 
-def load_emailgen_model(model_tag:str):
+
+def load_emailgen_model(model_tag: str):
     """
     load_emailgen_model - load a text generation pipeline for email generation
 
@@ -89,6 +90,7 @@ def load_emailgen_model(model_tag:str):
         device=0 if use_gpu else -1,
     )
 
+
 def get_parser():
     """
     get_parser - a helper function for the argparse module
@@ -115,7 +117,7 @@ def get_parser():
     )
 
     parser.add_argument(
-    "-nb",
+        "-nb",
         "--num_beams",
         type=int,
         default=4,
@@ -136,7 +138,11 @@ Hello,
 
 Following up on last week's bubblegum shipment, I"""
 
-available_models = ['postbot/distilgpt2-emailgen-V2', 'postbot/distilgpt2-emailgen', 'postbot/gpt2-medium-emailgen']
+available_models = [
+    "postbot/distilgpt2-emailgen-V2",
+    "postbot/distilgpt2-emailgen",
+    "postbot/gpt2-medium-emailgen",
+]
 
 if __name__ == "__main__":
     logging.info("\n\n\nStarting new instance of app.py")
@@ -206,8 +212,8 @@ if __name__ == "__main__":
                     value=model_tag,
                 )
                 load_model_button = gr.Button(
-                    'Load Model',
-                    variant='secondary',
+                    "Load Model",
+                    variant="secondary",
                 )
                 num_beams = gr.Radio(
                     choices=[4, 8, 12, 16],

utils.py

@@ -4,6 +4,7 @@
 import logging
 import re
 
+
 def postprocess(text: str):
     """
     postprocess - remove common values in scraped dataset