import streamlit as st
import pickle

with open('./trained_model.pkl', 'rb') as file:
    loaded_model = pickle.load(file)

input = st.text_area('Enter text')

if input:
    out = loaded_model(input)
    st.json(out[0][0])