Update app.py

app.py

@@ -106,10 +106,14 @@ print ("lm datasets")
 #Training Args
  
 training_args = TrainingArguments(
-    "test-tis",
+    output_dir="test-tis",
+    overwrite_output_dir = 'True',
     evaluation_strategy = "epoch",
     learning_rate=2e-5,
     weight_decay=0.01,
+    save_total_limit = 2
+    save_strategy = “no”
+    load_best_model_at_end=False
     #push_to_hub=True,
 )
 print ("training args")
@@ -120,7 +124,7 @@ trainer = Trainer(
         model=model,
         args=training_args,
         train_dataset=lm_datasets["train"],
-        eval_dataset=lm_datasets["train"],
+        eval_dataset=lm_datasets["test"],
         #compute_metrics=compute_metrics,
 )
 print ("trainer")