Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|