Ransaka commited on
Commit
112da8d
·
1 Parent(s): cea9d07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ st.markdown("The model used is [Ransaka/sinhala-bert-small](https://huggingface.
36
  # select model
37
  def get_model_id():
38
  st.subheader("Select a model to use")
39
- model_list = ["Ransaka/sinhala-bert-small","Ransaka/SinhalaRoberta","Ransaka/sinhala-bert-medium-v1"]#,"keshan/SinhalaBERTo"] #commented as tokenizer used here is differ from previous tow models.
40
  #TODO: Add "keshan/SinhalaBERTo" model as well
41
  selected_model = st.selectbox("Select Model", model_list)
42
  st.write(f"Selected model: {selected_model}")
 
36
  # select model
37
  def get_model_id():
38
  st.subheader("Select a model to use")
39
+ model_list = ["Ransaka/sinhala-bert-small","Ransaka/SinhalaRoberta","Ransaka/sinhala-bert-medium-v2","Ransaka/sinhala-bert-medium-v1"]#,"keshan/SinhalaBERTo"] #commented as tokenizer used here is differ from previous tow models.
40
  #TODO: Add "keshan/SinhalaBERTo" model as well
41
  selected_model = st.selectbox("Select Model", model_list)
42
  st.write(f"Selected model: {selected_model}")