jaimin commited on
Commit
c696da0
·
1 Parent(s): 1a2b328

Update styleformer.py

Browse files
Files changed (1) hide show
  1. styleformer.py +1 -0
styleformer.py CHANGED
@@ -11,6 +11,7 @@ class Styleformer():
11
  ):
12
  from transformers import AutoTokenizer
13
  from transformers import AutoModelForSeq2SeqLM
 
14
 
15
  self.style = style
16
  self.adequacy = adequacy_model_tag and Adequacy(model_tag=adequacy_model_tag, use_auth_token="access")
 
11
  ):
12
  from transformers import AutoTokenizer
13
  from transformers import AutoModelForSeq2SeqLM
14
+ from adequacy import Adequacy
15
 
16
  self.style = style
17
  self.adequacy = adequacy_model_tag and Adequacy(model_tag=adequacy_model_tag, use_auth_token="access")