Update app.py

app.py

@@ -4,6 +4,8 @@ from transformers import pipeline
 # transformers パイプラインのインポート
 fugu_translator_enja = pipeline('translation', model='staka/fugumt-en-ja')
 fugu_translator_jaen = pipeline('translation', model='staka/fugumt-ja-en')
+zhja_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-tc-big-zh-ja")
+jazh_translator = pipeline(model="larryvrh/mt5-translation-ja_zh")
 
 # デフォルトのモデル
 current_model = 'enja'
@@ -18,18 +20,22 @@ def translate(model, text):
     
     if current_model == 'enja':
         return fugu_translator_enja(text)[0]['translation_text']
-    else:
+    elif current_model == 'jaen':
         return fugu_translator_jaen(text)[0]['translation_text']
+    elif current_model == 'zhja':
+        return zhja_translator(text)[0]['translation_text']
+    elif current_model == 'jazh':
+        return jazh_translator(text)
 
 # Streamlit アプリケーション
-st.title("Fugu Translator")
+st.title("Multi-Language Translator")
 
 # デフォルトの入力値
 default_model = 'enja'
 default_text = ''
 
 # ユーザー入力の取得
-model = st.selectbox("モデル", ['enja', 'jaen'], index=0, key='model')
+model = st.selectbox("モデル", ['enja', 'jaen', 'zhja', 'jazh'], index=0, key='model')
 text = st.text_area("入力テキスト", default_text)
 
 # 翻訳ボタンが押されたときの処理