UI enhancements

apps/classifier.py

@@ -18,16 +18,23 @@ def load_model(input_text, model_name_or_path):
 
 
 def app():
- st.title("RoBERTa Marathi")
+ st.title("Marathi News Classifier")
 
  st.markdown(
- "This demo uses [RoBERTa for Marathi](https://huggingface.co/flax-community/roberta-base-mr) model "
- "trained on [mC4](https://huggingface.co/datasets/mc4)."
+ "This demo uses the below fine-tuned models for marathi news classification:\n"
+ "- [IndicNLP Marathi News Classifier](https://huggingface.co/flax-community/mr-indicnlp-classifier) fine-tuned on "
+ "[IndicNLP Marathi News Dataset](https://github.com/ai4bharat/indicnlp_corpus#indicnlp-news-article-classification-dataset)"
+ "\n"
+ "- [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.markdown("`IndicNLP Marathi News Dataset` consists of 3 classes - `['lifestyle', 'entertainment', 'sports']`")
+ st.markdown("`iNLTK Marathi News Dataset` consists of 3 classes - `['state', 'entertainment', 'sports']`")
+
  st.markdown(
- "Can't figure out where to get a sample text? Visit any of the following links, copy any headline and see if "
- "the model is predicting the respective class or not.\n"
+ "Can't figure out where to get a sample text other than the predefined ones? Visit any of the following links, "
+ "copy any headline and see if 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)\n"
@@ -36,25 +43,16 @@ def app():
 
  classifier = st.sidebar.selectbox("Select a Model", index=0, options=["IndicNLP", "iNLTK"])
 
- st.sidebar.markdown(
- "- [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 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"
+ "- lifestyle\n"
  "- entertainment\n"
  "- sports\n"
- "- lifestyle\n"
  "\n"
  "**iNLTK Classes**\n"
+ "- state\n"
  "- entertainment\n"
- "- sports\n"
- "- state"
+ "- sports"
  )
 
  sample_texts = [

apps/mlm.py

@@ -27,7 +27,7 @@ def app():
  )
 
  st.markdown(
- "Can't figure out where to get a sample text? Visit this "
+ "Can't figure out where to get a sample text other than the predefined ones? Visit this "
  "[link](https://maharashtratimes.com/entertainment/articlelist/19359255.cms), copy any headline and mask a word."
  )