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 = {
-        "Model_1": "pszemraj/long-t5-tglobal-base-16384-book-summary",
-        "Model_2": "pszemraj/long-t5-tglobal-base-sci-simplify",
-        "Model_3": "pszemraj/long-t5-tglobal-base-sci-simplify-elife",
-        "Model_4": "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
-        "Model_5": "pszemraj/pegasus-x-large-book-summary",
+        "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",
     }
     selected_model_identifier = MODEL_OPTIONS.get(model_name)
     device = "cuda" if torch.cuda.is_available() else "cpu"