MeetJivani commited on
Commit
16068a8
·
1 Parent(s): a759688

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -76,8 +76,8 @@ MODEL_OPTIONS = {
76
  "Model_5": "pszemraj/pegasus-x-large-book-summary",
77
  }
78
 
79
- model_names = list(MODEL_OPTIONS.keys()) # User-friendly labels
80
- model_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,8 +581,8 @@ if __name__ == "__main__":
581
  # label="Model Name",
582
  # )
583
  model_name = gr.Dropdown(
584
- choices=model_names,
585
- value=model_values[0],
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(