vives commited on
Commit
4ecc530
·
1 Parent(s): c770b72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ from transformers import AutoModelForMaskedLM
2
  from transformers import AutoTokenizer
3
  import streamlit as st
4
  model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
5
- model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
6
  text = st.text_area("enter some text!")
7
 
8
  if text:
9
- st.write(text)
 
2
  from transformers import AutoTokenizer
3
  import streamlit as st
4
  model_checkpoint = "vives/distilbert-base-uncased-finetuned-imdb-accelerate"
5
+ #model = AutoModelForMaskedLM.from_pretrained(model_checkpoint,output_hidden_states=True)
6
  text = st.text_area("enter some text!")
7
 
8
  if text:
9
+ st.write(len(text))