Create model.py
Browse files
model.py
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import AutoModel
|
| 2 |
+
|
| 3 |
+
model = AutoModel.from_pretrained("google-bert/bert-base-cased")
|
| 4 |
+
|
| 5 |
+
# Push the model to your namespace with the name "my-finetuned-bert".
|
| 6 |
+
model.push_to_hub("my-finetuned-bert")
|
| 7 |
+
|
| 8 |
+
# Push the model to an organization with the name "my-finetuned-bert".
|
| 9 |
+
model.push_to_hub("huggingface/model-test")
|