Upload 2 files

app.py

@@ -29,10 +29,10 @@ st.markdown("""
 """, unsafe_allow_html=True)
 
 # タイトルと説明
-st.title("💃 お嬢様語変換アプリ")
+st.title("💃 お嬢様語変換アプリ💃 ")
 st.markdown("""
 ごきげんよう！このアプリは、皆様の入力したテキストをエレガントなお嬢様語に変換いたしますわ！  
-テキストを入力して「変換するわ！」ボタンを押してくださいませ。おほほほほ！
+テキストを入力して「変換するわ！」ボタンを押してくださいませ。どんな言葉もお嬢様語に変換いたしますわ。おほほほほ！
 """)
 
 # サイドバーでエレガント度を設定
@@ -43,15 +43,20 @@ elegance_level = st.sidebar.slider("エレガント度", min_value=1, max_value=
 # プロンプトをエレガント度に応じて設定（日本語特化）
 prompt_templates = {
     1: "以下のテキストを丁寧で上品な日本語に変換してください。自然でエレガントな表現を使ってください：\n{}",
-    2: "以下のテキストをお嬢様風のエレガントな日本語に変換してください。「～ですわ」「おほほ」などを適度に使い、華やかで優雅な口調にしてください：\n{}",
-    3: "以下のテキストを非常に誇張されたお嬢様風の日本語に変換してください。「～ですわよ！」「おほほほほ！」を多用し、華麗でドラマチックな表現にしてください：\n{}"
+    2: "以下のテキストをお嬢様風のエレガントな日本語に変換してください。「～ですわ」「おほほ」などを適度に使い、華やかで優雅な口調にしてください。例: 「こんにちは」→「ごきげんよう、お元気ですわ？」：\n{}",
+    3: "以下のテキストを非常に誇張されたお嬢様風の日本語に変換してください。「～ですわよ！」「おほほほほ！」を多用し、華麗でドラマチックな表現にしてください。例: 「こんにちは」→「ごきげんようでございますわ！なんともお元気でいらっしゃいますの？おほほほほ！」：\n{}"
 }
 
 # LLMのロード（初回起動時にモデルをキャッシュ）
 @st.cache_resource
 def load_model():
     try:
-        return pipeline("text-generation", model="rinna/japanese-gpt2-small", device=-1)  # CPU使用
+        # rinna/japanese-gpt2-small をロード
+        pipe = pipeline("text-generation", model="rinna/japanese-gpt2-small", device=-1)  # CPU使用
+        # トークナイザーのパディング設定
+        if pipe.tokenizer.pad_token_id is None:
+            pipe.tokenizer.pad_token_id = pipe.tokenizer.eos_token_id
+        return pipe
     except Exception as e:
         st.error(f"モデルロードに失敗しましたわ！: {str(e)}")
         return None
@@ -59,7 +64,7 @@ def load_model():
 generator = load_model()
 
 # ユーザー入力
-input_text = st.text_input("テキストを入力してくださいませ（例: こんにちは、元気？）", "なんでや、阪神関係ないやろ！")
+input_text = st.text_input("テキストを入力してくださいませ（例: こんにちは、元気？）", "なんでや、阪神関係ないやろ - 入力: なんでや、阪神関係ないやろ！")
 
 # 変換ボタン
 if st.button("変換するわ！"):
@@ -74,12 +79,16 @@ if st.button("変換するわ！"):
                 # LLMでテキスト生成
                 try:
                     output = generator(prompt, max_length=150, num_return_sequences=1, 
-                                     do_sample=True, temperature=0.7, top_p=0.9, 
+                                     do_sample=True, temperature=0.9, top_p=0.95, 
                                      pad_token_id=generator.tokenizer.pad_token_id,
                                      bos_token_id=generator.tokenizer.bos_token_id,
                                      eos_token_id=generator.tokenizer.eos_token_id)
                     converted_text = output[0]["generated_text"].split(prompt)[-1].strip()
                     
+                    # 後処理：お嬢様らしさを強調
+                    if elegance_level >= 2 and not converted_text.endswith("おほほ"):
+                        converted_text += " おほほ！" if elegance_level == 2 else " おほほほほ！"
+                    
                     # 結果表示
                     st.subheader("変換結果")
                     st.markdown(f"**入力**: {input_text}")

requirements.txt

@@ -1,3 +1,4 @@
 streamlit==1.24.0
 transformers==4.44.2
-torch==2.0.1
+torch==2.0.1
+sentencepiece==0.2.0