hugging2021 commited on
Commit
5eefde1
·
verified ·
1 Parent(s): 96cf965

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,4 +1,7 @@
1
  import gradio as gr
 
 
 
2
 
3
  title = "BERT"
4
  description = "Gradio Demo for BERT. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
 
1
  import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ pipe = pipeline("fill-mask", model="google-bert/bert-base-uncased")
5
 
6
  title = "BERT"
7
  description = "Gradio Demo for BERT. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."