Debug

apps/inference.py

@@ -18,7 +18,7 @@ def load_model(masked_text, model_name):
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     MASK_TOKEN = tokenizer.mask_token
     masked_text = masked_text.replace("<mask>", MASK_TOKEN)
-    print(model_name, MASK_TOKEN, masked_text)
+    st.write(model_name, MASK_TOKEN, masked_text)
     nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
     result_sentence = nlp(masked_text)
     return result_sentence