Update app.py

app.py

@@ -643,7 +643,7 @@ def run_train_epoch(num):
      pro.init_training()
      for i in range(num):
     # model.train(True)
-        yield pro.run_train_epoch()
+        return  pro.run_train_epoch()
    else:
       pro.save_pretrained(pro.dir_model)
       pro.load_model()