Update app.py

app.py

@@ -42,10 +42,13 @@ if st.button("トレーニング開始"):
         tokenized_dataset = dataset.map(tokenize_function, batched=True)
         tokenized_dataset = tokenized_dataset.rename_column("text", "labels")
 
+        # 評価用データセットを使わない設定
+        eval_dataset = None  # 評価を行わない場合、eval_datasetはNoneに設定
+
         # トレーニング設定
         training_args = TrainingArguments(
             output_dir=output_dir,
-            eval_strategy="epoch",
+            evaluation_strategy="no",  # 評価を行わない設定
             learning_rate=2e-5,
             per_device_train_batch_size=8,
             num_train_epochs=1,
@@ -58,6 +61,7 @@ if st.button("トレーニング開始"):
             model=model,
             args=training_args,
             train_dataset=tokenized_dataset,
+            eval_dataset=eval_dataset,  # 評価データセットを指定しない
         )
 
         # トレーニングの実行