SinhalaMLM / app.py
Virajtharinda's picture
Update app.py
678cd5a verified
raw
history blame contribute delete
228 Bytes
import streamlit as st
from transformers import pipeline
# create pieline for MLM
pipe = pipeline('fill-mask', model = 'Virajtharinda/SinhalaWP')
text = st.text_area('enter some text!')
if text:
out = pipe(text)
st.json(out)