Spaces:
Runtime error
Runtime error
updated requirements.txt
Browse files
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,
|
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)
|