Erfan11 commited on
Commit
3e5ef5a
1 Parent(s): dc8dced

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -8
config.json CHANGED
@@ -1,8 +1,4 @@
1
- from transformers import AutoConfig, AutoModel
2
-
3
- # Load the configuration to find out what kind of model it is
4
- config = AutoConfig.from_pretrained('Erfan11/Neuracraft')
5
- print(config.model_type) # Check what model type it is
6
-
7
- # Load the model according to its type
8
- model = AutoModel.from_pretrained('Erfan11/Neuracraft')
 
1
+ {
2
+ "model_type": "bert",
3
+ "num_labels": 2
4
+ }