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
|