| import keras | |
| import joblib | |
| from happytransformer import HappyTextToText | |
| from IPython.display import FileLink | |
| from happytransformer import TTTrainArgs | |
| #args = TTTrainArgs(batch_size=8,num_train_epochs=10, learning_rate=1e-4) | |
| #happy_tt = HappyTextToText("T5", "prithivida/grammar_error_correcter_v1") | |
| #joblib.dump(happy_tt , 'finetuned_model_prithivida/grammar_error_correcter_v1') | |
| happy_tt = joblib.load("../input/grammar-error-correcter-model/grammar_error_correcter_v1_loss_0.28348055481910706lrt_1e-05") | |
| eval=[1.0] | |
| def gram(st): | |
| #happy_tt | |
| result_2 = m_jlib.generate_text(st, args=beam_settings) | |
| print(result_2.text) | |
| return result_2.text | |
| st=input() | |
| gram(st) |