jknebel commited on
Commit
1e6cafd
1 Parent(s): 0d3fc53

correct key error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def predict(prompt,model_name, max_length):
15
  summary = model(prompt,max_length)[0]["summary_text"]
16
  return summary
17
 
18
- options_1 = model_name_converter.keys()
19
  with gr.Blocks() as demo:
20
  drop_down = gr.Dropdown(choices=options_1, label="model")
21
  textbox = gr.Textbox(placeholder = "Enter text block to summarize", lines = 4)
 
15
  summary = model(prompt,max_length)[0]["summary_text"]
16
  return summary
17
 
18
+ options_1 = list(model_name_converter.keys())
19
  with gr.Blocks() as demo:
20
  drop_down = gr.Dropdown(choices=options_1, label="model")
21
  textbox = gr.Textbox(placeholder = "Enter text block to summarize", lines = 4)