MissingBreath commited on
Commit
ed35590
·
verified ·
1 Parent(s): a661456

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +3 -1
api.py CHANGED
@@ -9,7 +9,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
9
 
10
  # tokenizer = AutoTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
11
  from transformers import DistilBertTokenizer
12
- tokenizer = DistilBertTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
 
 
13
 
14
  model = AutoModelForSequenceClassification.from_pretrained("chillies/distilbert-course-review-classification")
15
 
 
9
 
10
  # tokenizer = AutoTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
11
  from transformers import DistilBertTokenizer
12
+
13
+ tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
14
+
15
 
16
  model = AutoModelForSequenceClassification.from_pretrained("chillies/distilbert-course-review-classification")
17