Update app.py

app.py

@@ -9,11 +9,11 @@ import os
 
 # Streamlit App
 st.title("Hugging Face Model Training App")
-st.write("castorini/afriberta-corpusを使って、ユーザーが入力したモデル名でファインチューニング")
+st.write("Nart/monolingual_abを使って、ユーザーが入力したモデル名でファインチューニング")
 
 # ユーザー入力
 model_name = st.text_input("トレーニングするモデル名 (例: Qwen/Qwen2.5-1.5B-Instruct)")
-dataset_name = "castorini/afriberta-corpus"
+dataset_name = "Nart/monolingual_ab"
 hf_token = st.text_input("Hugging Face Write トークン", type="password")
 repo_name = st.text_input("Hugging Faceリポジトリ名")  # ユーザーが入力できるリポジトリ名
 output_dir = "./finetuned_model"
@@ -51,10 +51,10 @@ if st.button("トレーニング開始"):
         training_args = TrainingArguments(
             output_dir=output_dir,
             eval_strategy="no",  # eval設定を評価しない設定
-            learning_rate=2e-5,
+            learning_rate=5e-5,
             per_device_train_batch_size=8,
             num_train_epochs=1,
-            save_steps=10_000,
+            save_steps=1_000,
             save_total_limit=2,
         )