Added classification model classes

app.py

@@ -5,7 +5,7 @@ from multiapp import MultiApp
 
 
 def main():
-    img = st.sidebar.image("images/huggingface-marathi-roberta.png", width=310)
+    img = st.sidebar.image("images/huggingface-marathi-roberta.png", width=230)
     app = MultiApp()
     app.add_app("Fill Mask", mlm.app)
     app.add_app("Text Classification", classifier.app)

apps/classifier.py

@@ -30,19 +30,32 @@ def app():
         "the model is predicting the respective class or not.\n"
         "- [entertainment](https://maharashtratimes.com/entertainment/articlelist/19359255.cms)\n"
         "- [sports](https://maharashtratimes.com/sports/articlelist/2429056.cms)\n"
-        "- [lifestyle](https://maharashtratimes.com/lifestyle-news/articlelist/2429025.cms)"
-        
+        "- [lifestyle](https://maharashtratimes.com/lifestyle-news/articlelist/2429025.cms)\n"
+        "- [state](https://maharashtratimes.com/maharashtra/articlelist/2429066.cms)"
     )
 
-    classifier = st.sidebar.selectbox("Select a Model", index=0, options=["Indic NLP", "iNLTK"])
+    classifier = st.sidebar.selectbox("Select a Model", index=0, options=["IndicNLP", "iNLTK"])
 
     st.sidebar.markdown(
-        "- [Indic NLP](https://huggingface.co/flax-community/mr-indicnlp-classifier) fine-tuned on "
+        "- [IndicNLP Marathi News Classifier](https://huggingface.co/flax-community/mr-indicnlp-classifier) fine-tuned on "
         "[Indic NLP Corpus for Marathi](https://github.com/ai4bharat/indicnlp_corpus#indicnlp-news-article-classification-dataset)"
         "\n"
-        "- [iNLTK](https://huggingface.co/flax-community/mr-inltk-classifier) fine-tuned on "
+        "- [iNLTK Marathi News Classifier](https://huggingface.co/flax-community/mr-inltk-classifier) fine-tuned on "
         "[Marathi News Dataset](https://www.kaggle.com/disisbig/marathi-news-dataset)"
     )
+    st.sidebar.markdown("\n")
+
+    st.sidebar.markdown(
+        "**IndicNLP Classes**\n"
+        "- entertainment\n"
+        "- sports\n"
+        "- lifestyle\n"
+        "\n"
+        "**iNLTK Classes**\n"
+        "- entertainment\n"
+        "- sports\n"
+        "- state"
+    )
 
     sample_texts = [
         "अध्यक्ष शरद पवार आणि उपमुख्यमंत्री अजित पवार यांची भेट घेतली.",

config.json

@@ -1,6 +1,6 @@
 {
     "models": {
-        "Indic NLP": "flax-community/mr-indicnlp-classifier",
+        "IndicNLP": "flax-community/mr-indicnlp-classifier",
         "iNLTK": "flax-community/mr-inltk-classifier",
         "RoBERTa": "flax-community/roberta-base-mr"
     }