Update app.py

app.py

@@ -31,12 +31,13 @@ SIN_2_ENG = {
 st.set_page_config(page_title="Sinhala zero-shot classification demo", page_icon=":bar_chart:")
 st.title("Sinhala zero-shot classification demo")
 st.markdown("This is a demo of the zero-shot classification pipeline from the [HuggingFace Transformers library](https://huggingface.co/transformers/).")
-st.markdown("The model used is [Ransaka/sinhala-bert-small](https://huggingface.co/Ransaka/sinhala-bert-small). However, you can use any model from the [HuggingFace model hub](https://huggingface.co/models).")
+st.markdown("The model used is [Ransaka/sinhala-bert-small](https://huggingface.co/Ransaka/sinhala-bert-small). However you can select a different model from the dropdown below.")
 
 # select model
 def get_model_id():
     st.subheader("Select a model to use")
-    model_list = ["Ransaka/sinhala-bert-small","Ransaka/SinhalaRoberta","keshan/SinhalaBERTo"]
+    model_list = ["Ransaka/sinhala-bert-small","Ransaka/SinhalaRoberta"]#,"keshan/SinhalaBERTo"] #commented as tokenizer used here is differ from previous tow models.
+    #TODO: Add "keshan/SinhalaBERTo" model as well
     selected_model = st.selectbox("Select Model", model_list)
     st.write(f"Selected model: {selected_model}")
     tokenizer = AutoTokenizer.from_pretrained(selected_model, token=hf_token)