Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 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")
+jazh_translator = pipeline(model="larryvrh/mt5-translation-ja_zh")
 
 # Streamlit アプリケーション
 st.title("Multi-Language Translator")
@@ -16,8 +16,8 @@ if 'session_models' not in st.session_state:
     st.session_state.session_models = {
         'enja': fugu_translator_enja,
         'jaen': fugu_translator_jaen,
-        'zhja': zhja_translator
-        #'jazh': jazh_translator
+        'zhja': zhja_translator,
+        'jazh': jazh_translator
     }
 
 # デフォルトの入力値
@@ -25,7 +25,7 @@ default_model = 'enja'
 default_text = ''
 
 # ユーザー入力の取得
-model = st.selectbox("モデル", ['enja', 'jaen', 'zhja'], index=0, key='model')
+model = st.selectbox("モデル", ['enja', 'jaen', 'zhja', 'jazh'], index=0, key='model')
 text = st.text_area("入力テキスト", default_text)
 
 # 翻訳ボタンが押されたときの処理
@@ -33,6 +33,9 @@ if st.button("翻訳する"):
     # Perform translation
     result = st.session_state.session_models[model](text)[0]['translation_text']
     
+    # Display the result
+    st.write(f"翻訳結果: {result}")
+    
     # Save the data to a CSV file
     data = {'ID': [1], 'Original Text': [text], 'Result': [result]}
     df = pd.DataFrame(data)
@@ -40,6 +43,3 @@ if st.button("翻訳する"):
     
     # Update the CSV creation flag
     st.session_state.csv_created = True
-    
-    # Display the result
-    st.write(f"翻訳結果: {result}")