Update summarize.py

summarize.py

@@ -21,11 +21,11 @@ def load_model_and_tokenizer(model_name: str) -> tuple:
     :return tuple: a tuple containing the model and tokenizer
     """
     MODEL_OPTIONS = {
-        "Text Summarization": "pszemraj/long-t5-tglobal-base-16384-book-summary",
-        "News Artical Summarization Alpha": "pszemraj/long-t5-tglobal-base-sci-simplify",
-        "News Artical Summarization Beta": "pszemraj/long-t5-tglobal-base-sci-simplify-elife",
-        "Scientific Document Summarization Alpha": "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
-        "Scientific Document Summarization Beta": "pszemraj/pegasus-x-large-book-summary",
+        "Text Summarizer": "pszemraj/long-t5-tglobal-base-16384-book-summary",
+        "News Artical Summarizer Alpha": "pszemraj/long-t5-tglobal-base-sci-simplify",
+        "News Artical Summarizer Beta": "pszemraj/long-t5-tglobal-base-sci-simplify-elife",
+        "Scientific Document Summarizer Alpha": "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
+        "Scientific Document Summarizer Beta": "pszemraj/pegasus-x-large-book-summary",
     }
     selected_model_identifier = MODEL_OPTIONS.get(model_name)
     device = "cuda" if torch.cuda.is_available() else "cpu"