hHoai commited on
Commit
4b9aa1d
·
verified ·
1 Parent(s): b793e16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
4
  import torch
5
 
6
  # Load mô hình
7
- model_checkpoint = "hHoai/bartPHO_final"
8
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
10
 
 
4
  import torch
5
 
6
  # Load mô hình
7
+ model_checkpoint = "hHoai/bartPHO_final/checkpoint-10026"
8
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
9
  model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
10