Update app.py

app.py

@@ -1,12 +1,11 @@
 import streamlit as st
 import pandas as pd
-from transformers import pipeline
+from transformers import pipelines
 
 # 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")
+fugu_translator_enja = pipelines.TranslationModel(model='staka/fugumt-en-ja')
+fugu_translator_jaen = pipelines.TranslationModel(model='staka/fugumt-ja-en')
+zhja_translator = pipelines.TranslationModel(model="Helsinki-NLP/opus-mt-tc-big-zh-ja")
 
 # Streamlit アプリケーション
 st.title("Multi-Language Translator")
@@ -16,16 +15,19 @@ 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
     }
 
+# 初期化
+if 'csv_created' not in st.session_state:
+    st.session_state.csv_created = False
+
 # デフォルトの入力値
 default_model = 'enja'
 default_text = ''
 
 # ユーザー入力の取得
-model = st.selectbox("モデル", ['enja', 'jaen', 'zhja', 'jazh'], index=0, key='model')
+model = st.selectbox("モデル", ['enja', 'jaen', 'zhja'], index=0, key='model')
 text = st.text_area("入力テキスト", default_text)
 
 # 翻訳ボタンが押されたときの処理