Update app.py

app.py

@@ -162,7 +162,8 @@ training_args = TrainingArguments(
     #fp16=True,
     save_strategy = "no",
     optim="adamw_torch",
-    load_best_model_at_end=False,
+    #load_best_model_at_end=False,
+    load_best_model_at_end=True
     #push_to_hub=True,
 )
 
@@ -220,8 +221,8 @@ print("Done Eval")
 #Push to Hub
 print("push to hub")
 login(token=os.environ["HF_WRITE_TOKEN"]) 
-trainer.push_to_hub("alexkueck/li-tis-tuned-1")
-#tokenizer.push_to_hub("test-tis-1")????
+#trainer.push_to_hub("alexkueck/li-tis-tuned-1")
+tokenizer.push_to_hub("alexkueck/li-tis-tuned-1")
 print("done")