:art: reformat

ai_single_response.py

@@ -24,7 +24,6 @@ from utils import print_spacer, cleantxt_wrap
 warnings.filterwarnings(action="ignore", message=".*gradient_checkpointing*")
 
 
-
 def extract_response(full_resp: list, plist: list, verbose: bool = False):
     """
     extract_response - helper fn for ai_single_response.py. By default aitextgen returns the prompt and the response, we just want the response
@@ -140,7 +139,6 @@ def query_gpt_model(
     """
     from aitextgen.utils import GPT2ConfigCPU
 
-
     ai = aitextgen(
         model="pszemraj/Ballpark-Trivia-L",  # THIS WORKS. XL is not working
         to_gpu=use_gpu,

app.py

@@ -157,7 +157,7 @@ if __name__ == "__main__":
         inputs=["text"],
         outputs="html",
         examples=[
-             "Who was the first king of France?",
+            "Who was the first king of France?",
             "What is the common name for a horse-drawn carriage used in city streets?",
             "What day of the week was November 8, 2016?",
             "What is Katy Perry's birthday?",
@@ -172,15 +172,15 @@ if __name__ == "__main__":
         "**Important Notes & About:**\n\n"
         "1. the model can take up to 60 seconds to respond sometimes, patience is a virtue.\n"
         "2. the model started from a pretrained checkpoint, and was trained on several different datasets. Anything it says should be fact-checked before being regarded as a true statement.\n "
-        "3. Some parameters are still being adjusted (in future, will have them as inpouts) any feedback is welcome :)\n"
+        "3. Some parameters are still being adjusted (in future, will have them as inpouts) any feedback is welcome :)\n",
         css="""
-        .chatbox {display:flex;flex-direction:column}
-        .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
-        .user_msg {background-color:cornflowerblue;color:white;align-self:start}
-        .resp_msg {background-color:lightgray;align-self:self-end}
-    """,
+            .chatbox {display:flex;flex-direction:column}
+            .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
+            .user_msg {background-color:cornflowerblue;color:white;align-self:start}
+            .resp_msg {background-color:lightgray;align-self:self-end}
+        """,
         allow_screenshot=True,
-        allow_flagging='never',
+        allow_flagging="never",
         theme="darkhuggingface",
     )