hHoai commited on
Commit
9ec8da8
·
verified ·
1 Parent(s): 3c6b99a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,7 +1,5 @@
1
  from model.model import Seq2SeqModel, predict, model_tokenize
2
 
3
-
4
-
5
  model_args = {
6
  "max_seq_length": 512,
7
  "max_length": 32,
@@ -10,7 +8,7 @@ model_args = {
10
 
11
  model = Seq2SeqModel(
12
  encoder_decoder_type="bartpho",
13
- encoder_decoder_name="/kaggle/input/model-1", #Checkpoint for model ASC-only for ACD-ASC, use detect_checkpoint-22415-epoch-5
14
  args=model_args,
15
  )
16
 
 
1
  from model.model import Seq2SeqModel, predict, model_tokenize
2
 
 
 
3
  model_args = {
4
  "max_seq_length": 512,
5
  "max_length": 32,
 
8
 
9
  model = Seq2SeqModel(
10
  encoder_decoder_type="bartpho",
11
+ encoder_decoder_name="result", #Checkpoint for model ASC-only for ACD-ASC, use detect_checkpoint-22415-epoch-5
12
  args=model_args,
13
  )
14