updated inference.py

apps/inference.py

@@ -13,12 +13,10 @@ def load_model(masked_text, model_name):
     from_flax = False
     if model_name == "flax-community/roberta-hindi":
         from_flax = True
-    # st.write(model_name, masked_text)
     model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=from_flax)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     MASK_TOKEN = tokenizer.mask_token
     masked_text = masked_text.replace("<mask>", MASK_TOKEN)
-    st.write(model_name, MASK_TOKEN, masked_text)
     nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
     result_sentence = nlp(masked_text)
     return result_sentence
@@ -48,7 +46,7 @@ def app():
         [
             "flax-community/roberta-hindi",
             "mrm8488/HindiBERTa",
-            "ai4bharat/indic-bert",
+            # "ai4bharat/indic-bert",
             "neuralspace-reverie/indic-transformers-hi-bert",
             "surajp/RoBERTa-hindi-guj-san",
         ],