Update app.py

app.py

@@ -168,7 +168,7 @@ print("trained!!!!!")
 
 
 ##################
-#Evaluate the new Model
+#Evaluate the new Model auf evual dataset
 print("Evaluate:")
 trainer.evaluate(eval_dataset=lm_datasets["test"])
 print("Done Eval")
@@ -180,12 +180,6 @@ login(token=os.environ["HF_WRITE_TOKEN"])
 trainer.save_model("test-tis-1")
 print("done")
 
-#######################################
-# Load model
-print("load model_neu")
-model_neu = trainer.load("test-tis-1")
-print("done load")
-
 #####################################
 #Push to Hub
 #print("push to hub")
@@ -194,6 +188,16 @@ print("done load")
 #tokenizer.push_to_hub("test-tis-1")????
 #print("done")
 
+#######################################
+# Load model
+print("load model_neu")
+model_name_neu = "alexkueck/test-tis-1"
+#model_neu = trainer.load("test-tis-1")
+
+#oder ...
+model_neu, device_neu = load_tokenizer_and_model(model_name_neu, False)
+print("done load")
+
 
 
 ############################