MissingBreath commited on
Commit
4330106
·
verified ·
1 Parent(s): a43c7d4

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +7 -7
api.py CHANGED
@@ -5,19 +5,19 @@ import io
5
  import tensorflow as tf
6
 
7
 
8
- # from transformers import AutoTokenizer, AutoModelForSequenceClassification
9
 
10
- # tokenizer = AutoTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
11
- # model = AutoModelForSequenceClassification.from_pretrained("chillies/distilbert-course-review-classification")
12
 
13
 
14
- from transformers import DistilBertTokenizerFast
15
 
16
- tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert-base-uncased")
17
 
18
- from transformers import pipeline
19
 
20
- model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
21
 
22
 
23
 
 
5
  import tensorflow as tf
6
 
7
 
8
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
9
 
10
+ tokenizer = AutoTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
11
+ model = AutoModelForSequenceClassification.from_pretrained("chillies/distilbert-course-review-classification")
12
 
13
 
14
+ # from transformers import DistilBertTokenizerFast
15
 
16
+ # tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert-base-uncased")
17
 
18
+ # from transformers import pipeline
19
 
20
+ # model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
21
 
22
 
23