xu-song commited on
Commit
1145ecb
·
verified ·
1 Parent(s): 1f8b36b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,12 +7,12 @@ from transformers import BertTokenizer, BertForMaskedLM
7
  import gradio as gr
8
  import time
9
 
10
- en_tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
11
- en_model = BertForMaskedLM.from_pretrained("bert-base-uncased")
12
  en_pipeline = PerplexityPipeline(model=en_model, tokenizer=en_tokenizer)
13
 
14
- zh_tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
15
- zh_model = BertForMaskedLM.from_pretrained("bert-base-chinese")
16
  zh_pipeline = PerplexityPipeline(model=zh_model, tokenizer=zh_tokenizer)
17
 
18
 
 
7
  import gradio as gr
8
  import time
9
 
10
+ en_tokenizer = BertTokenizer.from_pretrained('google-bert/bert-base-uncased')
11
+ en_model = BertForMaskedLM.from_pretrained("google-bert/bert-base-uncased")
12
  en_pipeline = PerplexityPipeline(model=en_model, tokenizer=en_tokenizer)
13
 
14
+ zh_tokenizer = BertTokenizer.from_pretrained('google-bert/bert-base-chinese')
15
+ zh_model = BertForMaskedLM.from_pretrained("google-bert/bert-base-chinese")
16
  zh_pipeline = PerplexityPipeline(model=zh_model, tokenizer=zh_tokenizer)
17
 
18