Jayaprakash commited on
Commit
bc5386a
·
1 Parent(s): d0829cd

create manage.py

Browse files
Files changed (1) hide show
  1. manage.py +18 -0
manage.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import keras
2
+ import joblib
3
+ from happytransformer import HappyTextToText
4
+ from IPython.display import FileLink
5
+ from happytransformer import TTTrainArgs
6
+ #args = TTTrainArgs(batch_size=8,num_train_epochs=10, learning_rate=1e-4)
7
+ #happy_tt = HappyTextToText("T5", "prithivida/grammar_error_correcter_v1")
8
+ #joblib.dump(happy_tt , 'finetuned_model_prithivida/grammar_error_correcter_v1')
9
+ happy_tt = joblib.load("../input/grammar-error-correcter-model/grammar_error_correcter_v1_loss_0.28348055481910706lrt_1e-05")
10
+ eval=[1.0]
11
+ def gram(st):
12
+ #happy_tt
13
+ result_2 = m_jlib.generate_text(st, args=beam_settings)
14
+ print(result_2.text)
15
+ return result_2.text
16
+
17
+ st=input()
18
+ gram(st)