feng2022 commited on
Commit
f8eb43e
1 Parent(s): 5303b5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,4 +1,9 @@
1
  import gradio as gr
 
 
 
 
 
2
  description = "BigGAN text-to-image demo."
3
  title = "BigGAN ImageNet"
4
  interface = gr.Interface.load("huggingface/osanseviero/BigGAN-deep-128",
 
1
  import gradio as gr
2
+ from transformers import BertTokenizer,BertModel
3
+ model_name = 'hfl/chinese-roberta-wwm-ext'
4
+ config = BertConfig.from_pretrained(model_name) # 这个方法会自动从官方的s3数据库下载模型配置、参数等信息(代码中已配置好位置)
5
+ tokenizer = BertTokenizer.from_pretrained(model_name) # 这个方法会自动从官方的s3数据库读取文件下的vocab.txt文件
6
+ model = BertModel.from_pretrained(model_name) # 这个方法会自动从官方的s3数据库下载模型信息
7
  description = "BigGAN text-to-image demo."
8
  title = "BigGAN ImageNet"
9
  interface = gr.Interface.load("huggingface/osanseviero/BigGAN-deep-128",