Added models info

app.py

@@ -1,11 +1,10 @@
-import streamlit as st
+from apps import about, mlm
 from multiapp import MultiApp
-from apps import about, inference
 
 
 def main():
     app = MultiApp()
-    app.add_app("Inference", inference.app)
+    app.add_app("Fill Mask", mlm.app)
     app.add_app("About", about.app)
     app.run()
 

apps/about.py

@@ -1,6 +1,8 @@
-import streamlit as st
-import os
 import json
+import os
+
+import streamlit as st
+
 
 def read_markdown(path, folder="./About/"):
     with open(os.path.join(folder, path)) as f:

apps/{inference.py → mlm.py}

@@ -1,12 +1,11 @@
-from pandas.io.formats.format import return_docstring
-import streamlit as st
-import pandas as pd
-from transformers import AutoTokenizer, AutoModelForMaskedLM
-from transformers import pipeline
-import os
 import json
+import os
 import random
 
+import pandas as pd
+import streamlit as st
+from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline
+
 with open("config.json") as f:
     cfg = json.loads(f.read())
 
@@ -27,13 +26,8 @@ def load_model(masked_text, model_name):
 
 
 def app():
-    st.markdown(
-        "<h1 style='text-align: center; color: green;'>RoBERTa Hindi</h1>",
-        unsafe_allow_html=True,
-    )
-    st.markdown(
-        "This demo uses multiple hindi transformer models for Masked Language Modelling (MLM)."
-    )
+    st.header("RoBERTa Hindi")
+    st.markdown("This demo uses multiple hindi transformer models for Masked Language Modelling (MLM).")
 
     models_list = list(cfg["models"].keys())
 
@@ -50,6 +44,14 @@ def app():
 
     results_df = pd.DataFrame(columns=["Model Name", "Filled Token", "Filled Text"])
 
+    st.sidebar.markdown(
+        "Models\n"
+        "- [RoBERTa Hindi](https://huggingface.co/flax-community/roberta-hindi)\n"
+        "- [Indic Transformers Hindi](https://huggingface.co/neuralspace-reverie/indic-transformers-hi-bert)\n"
+        "- [HindiBERTa]((https://huggingface.co/mrm8488/HindiBERTa)\n"
+        "- [RoBERTa Hindi Guj San](https://huggingface.co/surajp/RoBERTa-hindi-guj-san)"
+    )
+
     model_names = []
     filled_masked_texts = []
     filled_tokens = []
@@ -67,9 +69,7 @@ def app():
 
             for selected_model in models:
 
-                filled_sentence, filled_token = load_model(
-                    masked_text, cfg["models"][selected_model]
-                )
+                filled_sentence, filled_token = load_model(masked_text, cfg["models"][selected_model])
                 model_names.append(selected_model)
                 filled_tokens.append(filled_token)
                 filled_masked_texts.append(filled_sentence)

multiapp.py

@@ -1,5 +1,3 @@
-"""Frameworks for running multiple Streamlit applications as a single app.
-"""
 import streamlit as st
 
 
@@ -12,7 +10,6 @@ class MultiApp:
 
     def run(self):
         st.sidebar.header("Navigation")
-        app = st.sidebar.selectbox("", self.apps, format_func=lambda app: app["title"])
+        app = st.sidebar.radio("", self.apps, format_func=lambda app: app["title"])
 
         app["function"]()
-