Update app.py

app.py

@@ -6,9 +6,9 @@ import streamlit as st
 
 title = "Hokkien Translation"
 
-description = "Gradio Demo for fairseq S2S: speech-to-speech translation models. Select the model and record your input to translate between Hokkien and English"
+description = "Gradio Demo for fairseq speech-to-speech translation models. We have S2UT and UnitY models for bidirectional Hokkien and English translation. Please select the model and record your input to try it."
 
-article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2107.05604' target='_blank'>Speech-to-speech translation for a real-world unwritten language</a> | <a href='https://github.com/facebookresearch/fairseq/tree/ust/examples/hokkien' target='_blank'>Github Repo</a></p>"
+article = "<p style='text-align: center'><a href='https://ai.facebook.com/blog/ai-translation-hokkien/' target='_blank'>Speech-to-speech translation for a real-world unwritten language</a> | <a href='https://github.com/facebookresearch/fairseq/tree/ust/examples/hokkien' target='_blank'>Github Repo</a></p>"
 
 examples = [
   ["enhanced_direct_s2st_units_audios_es-en_set2_source_12478_cv.flac","xm_transformer_s2ut_800m-es-en-st-asr-bt_h1_2022"],
@@ -33,7 +33,7 @@ def inference(audio, model):
 
 gr.Interface(
     inference,
-    [gr.inputs.Audio(source="microphone", type="filepath", label="Input"),gr.inputs.Dropdown(choices=["xm_transformer_unity_en-hk", "xm_transformer_unity_en-hk"], default="xm_transformer_unity_en-hk",type="value", label="Model")
+    [gr.inputs.Audio(source="microphone", type="filepath", label="Input"),gr.inputs.Dropdown(choices=["xm_transformer_unity_en-hk", "xm_transformer_unity_en-hk", "xm_transformer_s2ut_en-hk", "xm_transformer_s2ut_hk-en"], default="xm_transformer_unity_en-hk",type="value", label="Model")
 ],
     gr.outputs.Audio(label="Output"),
     article=article,