Update app.py

app.py

@@ -13,14 +13,14 @@ def start_training(write_token, repo_name):
     dataset = load_dataset("range3/cc100-ja")
 
     # モデルとトークナイザーをロード
-    model_name = "Qwen/Qwen2.5-0.5B-Instruct"
+    model_name = "rinna/japanese-gpt-neox-small"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForCausalLM.from_pretrained(model_name)
 
     # トレーニング引数の設定
     training_args = TrainingArguments(
         output_dir="./results",          # 結果の保存先
-        num_train_epochs=3,              # エポック数
+        num_train_epochs=2,              # エポック数
         per_device_train_batch_size=8,   # バッチサイズ
         per_device_eval_batch_size=8,    # 評価バッチサイズ
         warmup_steps=500,                # ウォームアップステップ数