Spaces:
Sleeping
Sleeping
Explicit congif initialization
Browse files
app.py
CHANGED
@@ -8,9 +8,14 @@ import json
|
|
8 |
deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
|
9 |
deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC-RoBERTa", num_labels=19, args=deep_scc_model_args, use_cuda=False)
|
10 |
|
11 |
-
|
12 |
-
|
13 |
-
|
|
|
|
|
|
|
|
|
|
|
14 |
|
15 |
|
16 |
#Event Listeners
|
|
|
8 |
deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
|
9 |
deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC-RoBERTa", num_labels=19, args=deep_scc_model_args, use_cuda=False)
|
10 |
|
11 |
+
pragformer_config = transformers.AutoConfig.from_pretrained("Pragformer/PragFormer", trust_remote_code=True)
|
12 |
+
pragformer = transformers.AutoModel.from_pretrained("Pragformer/PragFormer", config=pragformer_config, trust_remote_code=True)
|
13 |
+
|
14 |
+
private_config = transformers.AutoConfig.from_pretrained("Pragformer/PragFormer_private", trust_remote_code=True)
|
15 |
+
pragformer_private = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_private", config=private_config, trust_remote_code=True)
|
16 |
+
|
17 |
+
reduction_config = transformers.AutoConfig.from_pretrained("Pragformer/PragFormer_reduction", trust_remote_code=True)
|
18 |
+
pragformer_reduction = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_reduction", config=reduction_config, trust_remote_code=True)
|
19 |
|
20 |
|
21 |
#Event Listeners
|