ThongCoding commited on
Commit
0df8cbf
·
verified ·
1 Parent(s): 635adca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,10 +7,10 @@ from transformers import MarianTokenizer, MarianMTModel, pipeline
7
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
 
9
  # Gọi API public
10
- vi2en_tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-vi-en", local_files_only=True)
11
- vi2en_model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-vi-en", local_files_only=True)
12
- en2vi_tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-vi", local_files_only=True)
13
- en2vi_model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-vi", local_files_only=True)
14
 
15
  translator_vi2en = pipeline("translation", model=vi2en_model, tokenizer=vi2en_tokenizer)
16
  translator_en2vi = pipeline("translation", model=en2vi_model, tokenizer=en2vi_tokenizer)
 
7
  client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
8
 
9
  # Gọi API public
10
+ vi2en_tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-vi-en")
11
+ vi2en_model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-vi-en")
12
+ en2vi_tokenizer = MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-vi")
13
+ en2vi_model = MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-vi")
14
 
15
  translator_vi2en = pipeline("translation", model=vi2en_model, tokenizer=vi2en_tokenizer)
16
  translator_en2vi = pipeline("translation", model=en2vi_model, tokenizer=en2vi_tokenizer)