Update app.py

app.py

@@ -61,13 +61,23 @@ nltk.download("punkt", force=True, quiet=True)
 nltk.download("popular", force=True, quiet=True)
 
 # Constants & Globals
-MODEL_OPTIONS = [
-    "pszemraj/long-t5-tglobal-base-16384-book-summary",
-    "pszemraj/long-t5-tglobal-base-sci-simplify",
-    "pszemraj/long-t5-tglobal-base-sci-simplify-elife",
-    "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
-    "pszemraj/pegasus-x-large-book-summary",
-]  # models users can choose from
+# MODEL_OPTIONS = [
+#     "pszemraj/long-t5-tglobal-base-16384-book-summary",
+#     "pszemraj/long-t5-tglobal-base-sci-simplify",
+#     "pszemraj/long-t5-tglobal-base-sci-simplify-elife",
+#     "pszemraj/long-t5-tglobal-base-16384-booksci-summary-v1",
+#     "pszemraj/pegasus-x-large-book-summary",
+# ]  # models users can choose from
+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",
+}
+
+model_names = list(MODEL_OPTIONS.keys())  # User-friendly labels
+model_values = list(MODEL_OPTIONS.values())  # Longer model names
 BEAM_OPTIONS = [2, 3, 4]  # beam sizes users can choose from
 TOKEN_BATCH_OPTIONS = [
     1024,
@@ -565,9 +575,14 @@ if __name__ == "__main__":
                 #     )
                 #     load_file_button = gr.Button("Upload & Process File")
                 with gr.Column(variant="compact"):
+                    # model_name = gr.Dropdown(
+                    #     choices=MODEL_OPTIONS,
+                    #     value=MODEL_OPTIONS[0],
+                    #     label="Model Name",
+                    # )
                     model_name = gr.Dropdown(
-                        choices=MODEL_OPTIONS,
-                        value=MODEL_OPTIONS[0],
+                        choices=model_names,
+                        value=model_names[0],
                         label="Model Name",
                     )
                     # example_name = gr.Dropdown(