t

app.py

@@ -292,7 +292,7 @@ All outputs generated are synthetic, created using fine-tuned models for experim
             gr.Markdown("Select a translation direction and enter text to translate between Dhivehi, English and Latin script.")
             # Set up initial choices and examples based on default model
             default_model = list(MODEL_OPTIONS_TRANSLATE.keys())[0]
-            if "MT5-B-Dhivehi-English" in default_model:
+            if default_model.startswith("MT"):
                 initial_choices = ["2dv", "2en"]
                 initial_value = "2dv"
             else:
@@ -337,7 +337,7 @@ All outputs generated are synthetic, created using fine-tuned models for experim
                 )
 
             def update_choices(model_name):
-                if "MT5-B-Dhivehi-English" in model_name:
+                if model_name.startswith("MT"):
                     return gr.update(choices=["2dv", "2en"], value="2dv")
                 else:
                     return gr.update(choices=["en2dv:", "dv2en:", "dv2latin:", "latin2dv:"], value="en2dv:")