Virajtharinda commited on
Commit
bd4d267
1 Parent(s): b921563

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
- # create pieline for MLM
4
 
5
- pipe = pipeline('fill-mask')
 
6
  text = st.text_area('enter some text!')
7
 
8
  if text:
 
1
  import streamlit as st
2
  from transformers import pipeline
 
3
 
4
+ # create pieline for MLM
5
+ pipe = pipeline('fill-mask', model = 'Ransaka/sinhala-bert-small')
6
  text = st.text_area('enter some text!')
7
 
8
  if text: