madankn79 commited on
Commit
ea8b822
·
1 Parent(s): f249f78
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,7 +22,6 @@ stop_words = set(stopwords.words("english")).union(extra_stopwords)
22
 
23
  # Model list
24
  model_choices = {
25
- "Xindus Fine-Tuned T5": "madankn/xindus_model_t5_base",
26
  "T5 Base (t5-base)": "t5-base",
27
  "DistilBART CNN (sshleifer/distilbart-cnn-12-6)": "sshleifer/distilbart-cnn-12-6",
28
  "DistilBART XSum (sshleifer/distilbart-xsum-12-6)": "sshleifer/distilbart-xsum-12-6",
@@ -121,7 +120,7 @@ iface = gr.Interface(
121
  fn=summarize_text,
122
  inputs=[
123
  gr.Textbox(lines=6, label="Enter text to summarize"),
124
- gr.Dropdown(choices=list(model_choices.keys()), label="Choose summarization model", value="Xindus Fine-Tuned T5"),
125
  gr.Slider(minimum=30, maximum=200, value=65, step=1, label="Max Character Limit")
126
  ],
127
  outputs=gr.Textbox(lines=3, label="Summary (truncated to character limit)"),
 
22
 
23
  # Model list
24
  model_choices = {
 
25
  "T5 Base (t5-base)": "t5-base",
26
  "DistilBART CNN (sshleifer/distilbart-cnn-12-6)": "sshleifer/distilbart-cnn-12-6",
27
  "DistilBART XSum (sshleifer/distilbart-xsum-12-6)": "sshleifer/distilbart-xsum-12-6",
 
120
  fn=summarize_text,
121
  inputs=[
122
  gr.Textbox(lines=6, label="Enter text to summarize"),
123
+ gr.Dropdown(choices=list(model_choices.keys()), label="Choose summarization model", value="T5 Base (t5-base)"),
124
  gr.Slider(minimum=30, maximum=200, value=65, step=1, label="Max Character Limit")
125
  ],
126
  outputs=gr.Textbox(lines=3, label="Summary (truncated to character limit)"),