RosiYo commited on
Commit
2a7bc10
·
1 Parent(s): 76a5c80

fix[bert]: Typo

Browse files
Files changed (1) hide show
  1. ml-service/model.py +2 -2
ml-service/model.py CHANGED
@@ -1,5 +1,5 @@
1
  # model.py
2
- from transformers import AutoModelForSequenceClassification, TFAutoModelForSequenceClassification, AutoTokenize
3
 
4
  class Model:
5
  """A model class to lead the model and tokenizer"""
@@ -12,5 +12,5 @@ class Model:
12
  return model
13
 
14
  def load_tokenizer():
15
- tokenizer = AutoTokenize.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
16
  return tokenizer
 
1
  # model.py
2
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
 
4
  class Model:
5
  """A model class to lead the model and tokenizer"""
 
12
  return model
13
 
14
  def load_tokenizer():
15
+ tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
16
  return tokenizer