Commit
•
75ea6ea
1
Parent(s):
46f5d6c
Create upload_model.py
Browse files- upload_model.py +9 -0
upload_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/my-finetuned-bert")
|