Update app.py

app.py

@@ -42,8 +42,8 @@ def main():
     training_args = TrainingArguments(
         output_dir="./output",
         evaluation_strategy="epoch",
-        per_device_train_batch_size=8,
-        per_device_eval_batch_size=8,
+        per_device_train_batch_size=1,
+        per_device_eval_batch_size=1,
         logging_dir="./logs",
         logging_steps=10,
         save_steps=100,                 # 每 100 步保存一次检查点