Update config.json
Browse files- config.json +4 -8
config.json
CHANGED
@@ -1,8 +1,4 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
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 |
+
}
|
|
|
|
|
|
|
|