Update app.py

app.py

@@ -20,6 +20,7 @@ def predict(text=None) -> dict:
     token_logits = model(input_ids, attention_mask=attention_mask).logits
     print(token_logits )
     mask_token_index = torch.where(inputs["input_ids"] == tokenizer.mask_token_id)[1]
+    print(tokenizer.mask_token_id, inputs["input_ids"])
     print(mask_token_index)
     mask_token_logits = token_logits[0, mask_token_index, :]
     print(mask_token_logits)