Upload app.py

app.py

@@ -40,18 +40,18 @@ st.sidebar.header("設定")
 elegance_level = st.sidebar.slider("エレガント度", min_value=1, max_value=3, value=2, step=1, 
                                   help="1: 控えめ, 2: 標準, 3: 超お嬢様")
 
-# プロンプトをエレガント度に応じて設定
+# プロンプトをエレガント度に応じて設定（日本語特化）
 prompt_templates = {
-    1: "以下のテキストを少しエレガントな日本語に変換してください。丁寧で上品な表現を使ってください：\n{}",
-    2: "以下のテキストをエレガントなお嬢様風の日本語に変換してください。華やかで優雅な表現で、「～ですわ」「おほほ」などを適度に使ってください：\n{}",
-    3: "以下のテキストを非常にエレガントで誇張されたお嬢様風の日本語に変換してください。華麗でドラマチックな表現で、「～ですわよ！」「おほほほほ！」を多用してください：\n{}"
+    1: "以下のテキストを丁寧で上品な日本語に変換してください。自然でエレガントな表現を使ってください：\n{}",
+    2: "以下のテキストをお嬢様風のエレガントな日本語に変換してください。「～ですわ」「おほほ」などを適度に使い、華やかで優雅な口調にしてください：\n{}",
+    3: "以下のテキストを非常に誇張されたお嬢様風の日本語に変換してください。「～ですわよ！」「おほほほほ！」を多用し、華麗でドラマチックな表現にしてください：\n{}"
 }
 
 # LLMのロード（初回起動時にモデルをキャッシュ）
 @st.cache_resource
 def load_model():
     try:
-        return pipeline("text-generation", model="gpt2", device=-1)  # CPU使用
+        return pipeline("text-generation", model="rinna/japanese-gpt2-small", device=-1)  # CPU使用
     except Exception as e:
         st.error(f"モデルロードに失敗しましたわ！: {str(e)}")
         return None
@@ -59,7 +59,7 @@ def load_model():
 generator = load_model()
 
 # ユーザー入力
-input_text = st.text_input("テキストを入力してくださいませ（例: こんにちは、元気？）", "こんにちは、元気？")
+input_text = st.text_input("テキストを入力してくださいませ（例: こんにちは、元気？）", "なんでや、阪神関係ないやろ！")
 
 # 変換ボタン
 if st.button("変換するわ！"):
@@ -74,7 +74,10 @@ 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.7, top_p=0.9, 
+                                     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()
                     
                     # 結果表示