Spaces:

govtech
/

off-topic-demo

Running

gabrielchua commited on 18 days ago

Commit

bcd67c9

verified ·

1 Parent(s): 1813258

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -210,7 +210,7 @@ STSB_MODEL_NAME = stsb_config['classifier']['embedding']['model_name']
 stsb_model_weights_fp = stsb_config['classifier']['embedding']['model_weights_fp']
 # Load STSB tokenizer and model
-stsb_tokenizer = AutoTokenizer.from_pretrained(STSB_MODEL_NAME)
 stsb_base_model = AutoModel.from_pretrained(STSB_MODEL_NAME)
 stsb_model = CrossEncoderWithMLP(stsb_base_model, num_labels=2)

 stsb_model_weights_fp = stsb_config['classifier']['embedding']['model_weights_fp']
 # Load STSB tokenizer and model
+stsb_tokenizer = AutoTokenizer.from_pretrained(STSB_MODEL_NAME, use_fast=False)
 stsb_base_model = AutoModel.from_pretrained(STSB_MODEL_NAME)
 stsb_model = CrossEncoderWithMLP(stsb_base_model, num_labels=2)