Anna-UoC commited on
Commit
dad1c14
·
1 Parent(s): 6d1d5ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ from transformers import AutoTokenizer
4
  from transformers import AutoModelForSequenceClassification
5
  from transformers import pipeline
6
 
7
- tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
8
 
9
- model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
10
  st.title("Text Classification with BERT")
11
 
12
  text = st.text_input("Enter some text")
 
4
  from transformers import AutoModelForSequenceClassification
5
  from transformers import pipeline
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
8
 
9
+ model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
10
  st.title("Text Classification with BERT")
11
 
12
  text = st.text_input("Enter some text")