File size: 563 Bytes
99886d2
800460f
99886d2
 
 
 
 
 
 
f1bee0d
 
99886d2
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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))