prateekagrawal commited on
Commit
4472bb8
·
1 Parent(s): 04d97c8

Debug indicberta

Browse files
Files changed (1) hide show
  1. apps/inference.py +1 -0
apps/inference.py CHANGED
@@ -18,6 +18,7 @@ def load_model(masked_text, model_name):
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  MASK_TOKEN = tokenizer.mask_token
20
  masked_text = masked_text.replace("<mask>", MASK_TOKEN)
 
21
  nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
22
  result_sentence = nlp(masked_text)
23
  return result_sentence
 
18
  tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  MASK_TOKEN = tokenizer.mask_token
20
  masked_text = masked_text.replace("<mask>", MASK_TOKEN)
21
+ print(model_name, MASK_TOKEN, masked_text)
22
  nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
23
  result_sentence = nlp(masked_text)
24
  return result_sentence