🎨

Signed-off-by: peter szemraj <peterszemraj@gmail.com>

app.py

@@ -179,7 +179,9 @@ if __name__ == "__main__":
 
     logging.info("Loading summ models")
     model, tokenizer = load_model_and_tokenizer("pszemraj/pegasus-x-large-book-summary")
-    model_sm, tokenizer_sm = load_model_and_tokenizer("pszemraj/long-t5-tglobal-base-16384-book-summary")
+    model_sm, tokenizer_sm = load_model_and_tokenizer(
+        "pszemraj/long-t5-tglobal-base-16384-book-summary"
+    )
 
     logging.info("Loading OCR model")
     with contextlib.redirect_stdout(None):

summarize.py

@@ -27,7 +27,9 @@ def load_model_and_tokenizer(model_name):
     return model, tokenizer
 
 
-def summarize_and_score(ids, mask, model, tokenizer, is_general_attention_model=True, **kwargs):
+def summarize_and_score(
+    ids, mask, model, tokenizer, is_general_attention_model=True, **kwargs
+):
     """
     summarize_and_score - given a batch of ids and a mask, return a summary and a score for the summary
 
@@ -79,7 +81,6 @@ def summarize_and_score(ids, mask, model, tokenizer, is_general_attention_model=
     return summary, score
 
 
-
 def summarize_via_tokenbatches(
     input_text: str,
     model,