Create app.py

app.py

@@ -0,0 +1,9 @@
+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)