Virajtharinda commited on
Commit
678cd5a
·
verified ·
1 Parent(s): bd4d267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ 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:
 
2
  from transformers import pipeline
3
 
4
  # create pieline for MLM
5
+ pipe = pipeline('fill-mask', model = 'Virajtharinda/SinhalaWP')
6
  text = st.text_area('enter some text!')
7
 
8
  if text: