xindus

app.py

@@ -22,7 +22,7 @@ stop_words = set(stopwords.words("english")).union(extra_stopwords)
 
 # Model list
 model_choices = {
-    "Xindus Fine-Tuned T5 (madankn/xindus_model_t5_base)": "madankn/xindus_model_t5_base"
+    "Xindus Fine-Tuned T5": "madankn/xindus_model_t5_base",
     "T5 Base (t5-base)": "t5-base",
     "DistilBART CNN (sshleifer/distilbart-cnn-12-6)": "sshleifer/distilbart-cnn-12-6",
     "DistilBART XSum (sshleifer/distilbart-xsum-12-6)": "sshleifer/distilbart-xsum-12-6",
@@ -121,7 +121,7 @@ iface = gr.Interface(
     fn=summarize_text,
     inputs=[
         gr.Textbox(lines=6, label="Enter text to summarize"),
-        gr.Dropdown(choices=list(model_choices.keys()), label="Choose summarization model", value="T5 Base (t5-base)"),
+        gr.Dropdown(choices=list(model_choices.keys()), label="Choose summarization model", value="Xindus Fine-Tuned T5"),
         gr.Slider(minimum=30, maximum=200, value=65, step=1, label="Max Character Limit")
     ],
     outputs=gr.Textbox(lines=3, label="Summary (truncated to character limit)"),