nouf-sst commited on
Commit
afa07cf
·
1 Parent(s): d7118d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ from torch.nn.utils.rnn import pad_sequence
16
  nlp = stanza.Pipeline(lang='en', processors='tokenize,pos,constituency')
17
  pos_tokenizer = AutoTokenizer.from_pretrained("QCRI/bert-base-multilingual-cased-pos-english")
18
  pos_model = AutoModelForTokenClassification.from_pretrained("QCRI/bert-base-multilingual-cased-pos-english")
19
- sentences_similarity_model = CrossEncoder('cross-encoder/stsb-roberta-base')
 
20
  nli_model = BertForSequenceClassification.from_pretrained("nouf-sst/bert-base-MultiNLI", use_auth_token="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")
21
  nli_tokenizer = BertTokenizer.from_pretrained("nouf-sst/bert-base-MultiNLI", use_auth_token="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", do_lower_case=True)
22
 
 
16
  nlp = stanza.Pipeline(lang='en', processors='tokenize,pos,constituency')
17
  pos_tokenizer = AutoTokenizer.from_pretrained("QCRI/bert-base-multilingual-cased-pos-english")
18
  pos_model = AutoModelForTokenClassification.from_pretrained("QCRI/bert-base-multilingual-cased-pos-english")
19
+ #sentences_similarity_model = CrossEncoder('cross-encoder/stsb-roberta-base')
20
+ sentences_similarity_model = CrossEncoder('WillHeld/roberta-base-stsb')
21
  nli_model = BertForSequenceClassification.from_pretrained("nouf-sst/bert-base-MultiNLI", use_auth_token="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")
22
  nli_tokenizer = BertTokenizer.from_pretrained("nouf-sst/bert-base-MultiNLI", use_auth_token="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", do_lower_case=True)
23