:art: organize imports and format

app.py

@@ -3,38 +3,36 @@ app.py - the main file for the app. This creates the flask app and handles the r
 
 """
 
-import torch
-from transformers import pipeline
-from cleantext import clean
-from pathlib import Path
-import warnings
-import time
 import argparse
 import logging
-import gradio as gr
-from gradio.inputs import Slider, Textbox
 import os
 import sys
+import time
+import warnings
 from os.path import dirname
+from pathlib import Path
+
+import gradio as gr
 import nltk
+import torch
+from cleantext import clean
+from gradio.inputs import Slider, Textbox
+from transformers import pipeline
+
 from converse import discussion
 from grammar_improve import (
+    build_symspell_obj,
     detect_propers,
+    fix_punct_spacing,
     load_ns_checker,
     neuspell_correct,
     remove_repeated_words,
     remove_trailing_punctuation,
-    build_symspell_obj,
     symspeller,
-    fix_punct_spacing,
-)
-
-from utils import (
-    cleantxt_wrap,
-    corr,
 )
+from utils import corr
 
-nltk.download("stopwords")  # TODO: find where this requirement originates from
+nltk.download("stopwords")  # download stopwords
 
 sys.path.append(dirname(dirname(os.path.abspath(__file__))))
 warnings.filterwarnings(action="ignore", message=".*gradient_checkpointing*")
@@ -197,9 +195,12 @@ if __name__ == "__main__":
     iface = gr.Interface(
         chat,
         inputs=[
-            Textbox(default_text="Why is everyone here eating chocolate cake?", label="prompt_message", placeholder="Enter a question",
-                    lines=2,
-                    ),
+            Textbox(
+                default="Why is everyone here eating chocolate cake?",
+                label="prompt_message",
+                placeholder="Enter a question",
+                lines=2,
+            ),
             Slider(
                 minimum=0.0, maximum=1.0, step=0.01, default=0.6, label="temperature"
             ),
@@ -219,7 +220,6 @@ if __name__ == "__main__":
             ["Are you single?", 0.6, 0.95, 138],
             ["Do you like people?", 0.7, 0.95, 25],
             ["You never took a short cut before?", 0.7, 0.95, 125],
-
         ],
         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"