Updated flax error in inference

apps/inference.py

@@ -15,7 +15,11 @@ predicted_sentence = []
 @st.cache(show_spinner=False, persist=True)
 def load_model(masked_text, model_name):
 
-    model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=True)
+    from_flax = False
+    if model_name == "flax-community/roberta-hindi":
+        from_flax = True
+
+    model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=from_flax)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
@@ -59,7 +63,7 @@ def app():
 
             for i in range(len(selected_models)):
                 filled_sentence = load_model(masked_text, selected_models[i])
-                st.write(filled_sentence)
+                # st.write(filled_sentence)
                 models.append(selected_models[i])
                 predicted_tokens.append(filled_sentence[0]["token_str"])
                 predicted_sentence.append(filled_sentence[0]["sequence"])