Replaced random checkbox

README.md

@@ -1,8 +1,8 @@
 ---
-title: Roberta Testing
-emoji: 📉
+title: Roberta Hindi
+emoji: 📚
 colorFrom: green
-colorTo: indigo
+colorTo: gray
 sdk: streamlit
 app_file: app.py
 pinned: false

apps/inference.py

@@ -37,7 +37,7 @@ def app():
 
     models_list = list(cfg["models"].keys())
 
-    models = st.multiselect("Choose models", models_list, models_list[0],)
+    models = st.multiselect("Choose models", models_list, models_list)
 
     target_text_path = "./mlm_custom/mlm_targeted_text.csv"
     target_text_df = pd.read_csv(target_text_path)
@@ -46,14 +46,12 @@ def app():
 
     st.sidebar.title("Hindi MLM")
 
-    pick_random = st.sidebar.checkbox("Pick any random text")
-
     results_df = pd.DataFrame(columns=["Model Name", "Filled Token", "Filled Text"])
 
     model_names = []
     filled_masked_texts = []
     filled_tokens = []
-
+    pick_random = st.checkbox("Pick any random text")
     if pick_random:
         random_text = texts[random.randint(0, texts.shape[0] - 1)]
         masked_text = st.text_area("Please type a masked sentence to fill", random_text)

mlm_custom/mlm_test_config.csv

@@ -3,4 +3,4 @@ flax-community/roberta-hindi,,,TRUE,TRUE,TRUE
 mrm8488/HindiBERTa,,,FALSE,TRUE,TRUE
 ai4bharat/indic-bert,,,FALSE,FALSE,FALSE
 neuralspace-reverie/indic-transformers-hi-bert,,,FALSE,TRUE,TRUE
-surajp/RoBERTa-hindi-guj-san,,,FALSE,TRUE,TRUE
+surajp/RoBERTa-hindi-guj-san,,,FALSE,TRUE,TRUE

mlm_custom/test_mlm.py

@@ -1,9 +1,10 @@
-import pandas as pd
-import numpy as np
-from transformers import AutoTokenizer, RobertaModel, AutoModel, AutoModelForMaskedLM
-from transformers import pipeline
-import os
 import json
+import os
+
+import numpy as np
+import pandas as pd
+from transformers import (AutoModel, AutoModelForMaskedLM, AutoTokenizer,
+                          RobertaModel, pipeline)
 
 
 class MLMTest():

multiapp.py

@@ -11,7 +11,7 @@ class MultiApp:
         self.apps.append({"title": title, "function": func})
 
     def run(self):
-        # st.sidebar.header("Navigation")
+        st.sidebar.header("Navigation")
         app = st.sidebar.selectbox("", self.apps, format_func=lambda app: app["title"])
 
         app["function"]()

requirements.txt

@@ -1,6 +1,3 @@
 streamlit
 torch
-transformers[sentencepiece]
-jax
-jaxlib
-flax
+transformers