Hugging Face's logo

Spaces:
ithuan
/
formosan-translation
Sleeping

App Files Community
1
txya900619 commited on
Commit
90666ff
·
verified ·
1 Parent(s): 7ed3770

feat: use v2 model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ FORMOSAN_LANGUAGES_MAP = {
50
 
51
  ETHNICITIES = list(set([k.split("_")[0] for k in FORMOSAN_LANGUAGES_MAP.keys()]))
52
 
53
- MODEL_NAME = "ithuan/nllb-600m-formosan-all-finetune"
54
  device = "cuda" if torch.cuda.is_available() else "cpu"
55
 
56
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)
 
50
 
51
  ETHNICITIES = list(set([k.split("_")[0] for k in FORMOSAN_LANGUAGES_MAP.keys()]))
52
 
53
+ MODEL_NAME = "ithuan/nllb-600m-formosan-all-finetune-v2"
54
  device = "cuda" if torch.cuda.is_available() else "cpu"
55
 
56
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME).to(device)