Updated inference.py

apps/inference.py

@@ -35,6 +35,10 @@ def app():
         "This demo uses multiple hindi transformer models for Masked Language Modelling (MLM)."
     )
 
+    models_list = list(cfg["models"].keys())
+
+    models = st.multiselect("Choose models", models_list, models_list[0],)
+
     target_text_path = "./mlm_custom/mlm_targeted_text.csv"
     target_text_df = pd.read_csv(target_text_path)