Hugging Face's logo

Spaces:
hHoai
/
vietnam_correction
Runtime error

App Files Community
hHoai commited on
Commit
bf765f3
·
verified ·
1 Parent(s): 4b9aa1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,8 +4,9 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
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
 
11
  def spellcheck(text):
 
4
  import torch
5
 
6
  # Load mô hình
7
+ model_checkpoint = "hHoai/bartPHO_final"
8
+ tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, revision="checkpoint-10026")
9
+
10
  model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
11
 
12
  def spellcheck(text):