supplied revision

app.py

@@ -8,9 +8,9 @@ import json
 deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
 deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC-RoBERTa", num_labels=19, args=deep_scc_model_args, use_cuda=False)
 
-pragformer = transformers.AutoModel.from_pretrained("Pragformer/PragFormer", trust_remote_code=True)
-pragformer_private = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_private", trust_remote_code=True)
-pragformer_reduction = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_reduction", trust_remote_code=True)
+pragformer = transformers.AutoModel.from_pretrained("Pragformer/PragFormer", revision="main", trust_remote_code=True)
+pragformer_private = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_private", revision="main", trust_remote_code=True)
+pragformer_reduction = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_reduction", revision="main", trust_remote_code=True)
 
 
 #Event Listeners