MendoBERT_RE / app.py
Shiro26's picture
Update app.py
f1bee0d
raw
history blame
563 Bytes
import streamlit as st
from transformers import pipeline
model = pipeline("text-classification", model="/home/user/app/MendoBERT/", tokenizer="indolem/indobert-base-uncased")
basemodel = pipeline("text-classification", model="/home/user/app/IndoLEM/", tokenizer="indolem/indobert-base-uncased")
if 'options' not in st.session_state:
st.session_state['options'] = ""
placeholder = st.empty()
with placeholder:
text = st.text_area('Enter some text: ', key = 'options')
if text:
st.write(model(text))
st.write("\n")
st.write(basemodel(text))