Update app.py

app.py

@@ -359,15 +359,16 @@ print("trained!!!!!")
 
 ##################
 #Evaluate the new Model auf evual dataset
-print ("################################")
-print("Evaluate:")
-trainer.evaluate(eval_dataset=lm_datasets["test"])
+print ("#################Evaluate:#################")
+loss, accuracy = trainer.evaluate(eval_dataset=lm_datasets["test"])
+print(loss, accuracy)
 print("Done Eval")
 
 print('################ Test Trained Model ###################')
-#predictions = trainer.predict(lm_datasets["test"])
-#preds = np.argmax(predictions.predictions, axis=-1)
-
+predictions = trainer.predict(lm_datasets["test"])
+preds = np.argmax(predictions.predictions, axis=-1)
+print('###########preds##################')
+print(preds)
 
 
 ###################################################