app.py

from transformers import AutoModelForMaskedLM
from transformers import AutoTokenizer
import streamlit as st
model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
text = st.text_area("enter some text!")

if text:
  st.json(text)