Owos commited on
Commit
f82b5e5
·
1 Parent(s): 89d9abb

updated requirements.txt

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -197,7 +197,7 @@ if run:
197
  MLM_UNK_TOKEN = tokenizer.unk_token_id
198
 
199
  BATCH_SIZE = 1
200
- score = get_sense_score_batched(sentence,tokenizer,model,MLM_MASK_TOKEN,MLM_UNK_TOKEN,None,BATCH_SIZE)
201
  scores[model_id] = score
202
  my_bar.progress(index + 1, text=progress_text)
203
  scores = sort_dictionary(scores)
 
197
  MLM_UNK_TOKEN = tokenizer.unk_token_id
198
 
199
  BATCH_SIZE = 1
200
+ score = get_sense_score_batched(sentence,tokenizer,model,MLM_MASK_TOKEN,MLM_UNK_TOKEN,BATCH_SIZE)
201
  scores[model_id] = score
202
  my_bar.progress(index + 1, text=progress_text)
203
  scores = sort_dictionary(scores)