Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 available_models = {
-    "mt5.baseline": pipeline("text2text-generation", model="samzirbo/mt5.baseline"),
-    "mt5.genered": pipeline("text2text-generation", model="samzirbo/mt5.gendered"),
-    "mt5.balanced": pipeline("text2text-generation", model="samzirbo/mt5.balanced"),
-    "mt5.gendered_balanced": pipeline("text2text-generation", model="samzirbo/mt5.gendered_balanced")
+    "Baseline": pipeline("text2text-generation", model="samzirbo/mt5.baseline"),
+    "Genered": pipeline("text2text-generation", model="samzirbo/mt5.gendered"),
+    "Balanced": pipeline("text2text-generation", model="samzirbo/mt5.balanced"),
+    "Gendered and Balanced": pipeline("text2text-generation", model="samzirbo/mt5.gendered_balanced")
 }
 
 def translate_text(model_name, lang_dir, gender, input_text):
@@ -21,14 +21,13 @@ def translate_text(model_name, lang_dir, gender, input_text):
     return output_text[0]['generated_text']
 
 
-model_dropdown = gr.Dropdown(choices=list(available_models.keys()), label="Select Model", value="mt5.baseline")
+model_dropdown = gr.Dropdown(choices=list(available_models.keys()), label="Select Model", value="Baseline")
 lang_dropdown = gr.Dropdown(choices=["English -> Spanish", "Spanish -> English"], label="Language Direction", value="English -> Spanish")
 gender_dropdown = gr.Dropdown(choices=["female", "male"], label="Select Gender", value=None)
 
 iface = gr.Interface(fn=translate_text, 
                      inputs=[model_dropdown, lang_dropdown, gender_dropdown, "text"], 
                      outputs="text", 
-                     title="Translation Interface",
-                     description="Select a model, language direction, and input text to translate.")
+                     title="Translation Interface")
 
 iface.launch()