:art: format to black

app.py

@@ -57,8 +57,13 @@ def chat(trivia_query, temperature=0.7, top_p=0.95, top_k=50):
         [str]: the bot's response
     """
     history = []
-    response = ask_gpt(message=trivia_query, chat_pipe=my_chatbot,
-                       top_p=top_p, top_k=top_k, temperature=temperature)
+    response = ask_gpt(
+        message=trivia_query,
+        chat_pipe=my_chatbot,
+        top_p=top_p,
+        top_k=top_k,
+        temperature=temperature,
+    )
     history = [trivia_query, response]
     html = ""
     for item in history:
@@ -152,7 +157,7 @@ def get_parser():
     parser.add_argument(
         "--basic-sc",
         required=False,
-        default=True, # TODO: change this back to False once Neuspell issues are resolved.
+        default=True,  # TODO: change this back to False once Neuspell issues are resolved.
         action="store_true",
         help="turn on symspell (baseline) correction instead of the more advanced neural net models",
     )
@@ -191,17 +196,21 @@ if __name__ == "__main__":
     print(f"using model stored here: \n {model_loc} \n")
     iface = gr.Interface(
         chat,
-        inputs=["text",
-                Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"),
-                Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
-                Slider(minimum=0, maximum=250, step=1, default=50, label="top_k")],
+        inputs=[
+            "text",
+            Slider(
+                minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"
+            ),
+            Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="top_p"),
+            Slider(minimum=0, maximum=250, step=1, default=50, label="top_k"),
+        ],
         outputs="html",
         examples_per_page=8,
         examples=[
             ["How can you help me?", 0.6, 0.95, 50],
             ["what can you do?", 0.6, 0.95, 50],
             ["Hi, my name is……", 0.6, 0.95, 50],
-            ["Happy birthday!",     0.9, 0.95, 50],
+            ["Happy birthday!", 0.9, 0.95, 50],
             ["I have a question, can you help me?", 0.6, 0.95, 50],
             ["Do you know a joke?", 0.6, 0.85, 50],
             ["Will you marry me?", 0.6, 0.95, 138],
@@ -211,11 +220,10 @@ if __name__ == "__main__":
             ["Do you have a hobby?", 0.6, 0.95, 50],
             ["You’re clever", 0.6, 0.95, 50],
             ["Tell me about your personality", 0.6, 0.95, 50],
-            ["You’re annoying",     0.6, 0.95, 50],
+            ["You’re annoying", 0.6, 0.95, 50],
             ["I want to speak to a human now.", 0.6, 0.95, 50],
-            ["Don’t you speak English?!",   0.6, 0.95, 50],
+            ["Don’t you speak English?!", 0.6, 0.95, 50],
             ["Are you human?", 0.6, 0.95, 50],
-
         ],
         title=f"GPT Chatbot Demo: {default_model} Model",
         description=f"A Demo of a Chatbot trained for conversation with humans. Size XL= 1.5B parameters.\n\n"

converse.py

@@ -213,8 +213,12 @@ def consolidate_texts(
     name_spk = "person alpha" if name_spk is None else name_spk
     if verbose:
         print("====" * 10)
-        print(f"\n[DEBUG] initial model_resp has {len(model_resp)} lines: \n\t{model_resp}")
-        print(f" the first element is \n\t{model_resp[0]} and it is {type(model_resp[0])}")
+        print(
+            f"\n[DEBUG] initial model_resp has {len(model_resp)} lines: \n\t{model_resp}"
+        )
+        print(
+            f" the first element is \n\t{model_resp[0]} and it is {type(model_resp[0])}"
+        )
     fn_resp = []
 
     name_counter = 0
@@ -228,7 +232,10 @@ def consolidate_texts(
             if print_debug:
                 print(f"\nDEBUG: \n\t{resline}\ncaused the break")
             break  # the name of the speaker is in the line, so we're done
-        if any([": " in resline,":\n" in resline]) and name_resp.lower() not in resline.lower():
+        if (
+            any([": " in resline, ":\n" in resline])
+            and name_resp.lower() not in resline.lower()
+        ):
             if print_debug:
                 print(f"\nDEBUG: \n\t{resline}\ncaused the break")
             break