Spaces:
Runtime error
Runtime error
Commit
·
16068a8
1
Parent(s):
a759688
Update app.py
Browse files
app.py
CHANGED
@@ -76,8 +76,8 @@ MODEL_OPTIONS = {
|
|
76 |
"Model_5": "pszemraj/pegasus-x-large-book-summary",
|
77 |
}
|
78 |
|
79 |
-
|
80 |
-
|
81 |
BEAM_OPTIONS = [2, 3, 4] # beam sizes users can choose from
|
82 |
TOKEN_BATCH_OPTIONS = [
|
83 |
1024,
|
@@ -581,8 +581,8 @@ if __name__ == "__main__":
|
|
581 |
# label="Model Name",
|
582 |
# )
|
583 |
model_name = gr.Dropdown(
|
584 |
-
choices=
|
585 |
-
value=
|
586 |
label="Model Name",
|
587 |
)
|
588 |
# example_name = gr.Dropdown(
|
|
|
76 |
"Model_5": "pszemraj/pegasus-x-large-book-summary",
|
77 |
}
|
78 |
|
79 |
+
MODAL_NAMES = list(MODEL_OPTIONS.keys()) # User-friendly labels
|
80 |
+
MODAL_VALUES = list(MODEL_OPTIONS.values()) # Longer model names
|
81 |
BEAM_OPTIONS = [2, 3, 4] # beam sizes users can choose from
|
82 |
TOKEN_BATCH_OPTIONS = [
|
83 |
1024,
|
|
|
581 |
# label="Model Name",
|
582 |
# )
|
583 |
model_name = gr.Dropdown(
|
584 |
+
choices=MODAL_NAMES,
|
585 |
+
value=MODAL_VALUES[0],
|
586 |
label="Model Name",
|
587 |
)
|
588 |
# example_name = gr.Dropdown(
|