NeerAbhy commited on
Commit
40c7b63
·
verified ·
1 Parent(s): 3881c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -229,7 +229,7 @@ with demo:
229
  text = gr.Textbox()
230
  #gr.Markdown("Speech analyzer")
231
  #audio = gr.Audio(type="filepath", label = "Upload a file")
232
- model_dropdown = gr.Dropdown(choices=model_names, label="Choose a model", value="sshleifer/distilbart-cnn-12-6")
233
  load_message = gr.Textbox(label="Load Status", interactive=False)
234
  b1 = gr.Button("Load Model")
235
  min_length_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Minimum Summary Length (%)", value=10)
@@ -237,7 +237,7 @@ with demo:
237
 
238
  summarize_button = gr.Button("Summarize Text")
239
 
240
- b1.click(fn=load_model, inputs=model_dropdown, outputs=load_message)
241
  summarize_button.click(fn=summarize_text, inputs=[text0, min_length_slider, max_length_slider],
242
  outputs=text)
243
 
 
229
  text = gr.Textbox()
230
  #gr.Markdown("Speech analyzer")
231
  #audio = gr.Audio(type="filepath", label = "Upload a file")
232
+ model_dropdown = gr.Dropdown(choices = summarization_model_names, label="Choose a model", value="sshleifer/distilbart-cnn-12-6")
233
  load_message = gr.Textbox(label="Load Status", interactive=False)
234
  b1 = gr.Button("Load Model")
235
  min_length_slider = gr.Slider(minimum=0, maximum=100, step=1, label="Minimum Summary Length (%)", value=10)
 
237
 
238
  summarize_button = gr.Button("Summarize Text")
239
 
240
+ b1.click(fn=load_summarization_model, inputs=model_dropdown, outputs=load_message)
241
  summarize_button.click(fn=summarize_text, inputs=[text0, min_length_slider, max_length_slider],
242
  outputs=text)
243