cccc commited on
Commit
1820058
·
1 Parent(s): dd6d97a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- unmasker = pipeline('fill-mask', model='roberta-base') #bert-base-multilingual-cased') #bert-base-uncased')
5
 
6
  def fill_mask(text):
7
  unmasked = unmasker(text)
@@ -79,4 +79,4 @@ footer {display:none !important}
79
  """
80
 
81
  demo = gr.Interface(fn=fill_mask, inputs=gr.Textbox(lines=1, label="Input"), outputs=gr.Label(label="Output"),title="Fill Mask | Data Science Dojo", theme="light", examples=examples, css=css)
82
- demo.launch()
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ unmasker = pipeline('fill-mask', model='bert-base-uncased')
5
 
6
  def fill_mask(text):
7
  unmasked = unmasker(text)
 
79
  """
80
 
81
  demo = gr.Interface(fn=fill_mask, inputs=gr.Textbox(lines=1, label="Input"), outputs=gr.Label(label="Output"),title="Fill Mask | Data Science Dojo", theme="light", examples=examples, css=css)
82
+ demo.launch()