DS-20202 commited on
Commit
4d5d4a7
·
1 Parent(s): 7d0e10e

Downloading models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ bert_original = pipeline('fill-mask', model='Daniel-Saeedi/Sent-Debias-bert-gend
10
 
11
  def fill_mask(stmt,model):
12
  if model == 'bert':
13
- return bert_debiased(stmt), bert_original(stmt)
14
 
15
 
16
  demo = gr.Interface(
 
10
 
11
  def fill_mask(stmt,model):
12
  if model == 'bert':
13
+ return bert_debiased(stmt), bert_original(stmt), ''
14
 
15
 
16
  demo = gr.Interface(