🎨 format

Signed-off-by: peter szemraj <[email protected]>

converse.py

@@ -5,7 +5,10 @@
 """
 
 import logging
-logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+)
 import pprint as pp
 import time
 
@@ -13,6 +16,7 @@ from grammar_improve import remove_trailing_punctuation
 
 from constrained_generation import constrained_generation
 
+
 def discussion(
     prompt_text: str,
     speaker: str,
@@ -65,10 +69,9 @@ def discussion(
     if verbose:
         print("overall prompt:\n")
         pp.pprint(this_prompt, indent=4)
-    # call the model
-    print("\n... generating...")
+
     if constrained_beam_search:
-        logging.info("using constrained beam search")
+        logging.info("generating using constrained beam search ...")
         response = constrained_generation(
             prompt=this_prompt,
             pipeline=pipeline,
@@ -85,15 +88,13 @@ def discussion(
 
         bot_dialogue = consolidate_texts(
             name_resp=responder,
-            model_resp=response.split(
-        "\n"
-    ),
+            model_resp=response.split("\n"),
             name_spk=speaker,
             verbose=verbose,
             print_debug=True,
         )
     else:
-        logging.info("using sampling")
+        logging.info("generating using sampling ...")
         bot_dialogue = gen_response(
             this_prompt,
             pipeline,
@@ -140,15 +141,15 @@ def gen_response(
     speaker: str,
     responder: str,
     timeout=45,
-    min_length=4,
+    min_length=12,
     max_length=48,
     top_p=0.95,
     top_k=20,
     temperature=0.5,
     full_text=False,
     num_return_sequences=1,
-    length_penalty:float=0.8,
-    repetition_penalty:float=3.5,
+    length_penalty: float = 0.8,
+    repetition_penalty: float = 3.5,
     no_repeat_ngram_size=2,
     device=-1,
     verbose=False,