Spaces:
Runtime error
Runtime error
Commit
·
a2d41fa
1
Parent(s):
7945703
Update trainer.py
Browse files- trainer.py +13 -7
trainer.py
CHANGED
|
@@ -75,13 +75,19 @@ model.evaluate(test_images, test_labels)
|
|
| 75 |
|
| 76 |
# save the model
|
| 77 |
model.save("model.keras")
|
| 78 |
-
push_to_hub_keras
|
| 79 |
-
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 85 |
)
|
| 86 |
|
| 87 |
|
|
|
|
| 75 |
|
| 76 |
# save the model
|
| 77 |
model.save("model.keras")
|
| 78 |
+
#from transformers import push_to_hub_keras
|
| 79 |
+
|
| 80 |
+
# Save the model
|
| 81 |
+
#model.save("model.keras")
|
| 82 |
+
|
| 83 |
+
# Upload the model to your Hugging Face space repository
|
| 84 |
+
push_to_hub_keras(
|
| 85 |
+
model,
|
| 86 |
+
repository_name="okeowo1014/trainingsample",
|
| 87 |
+
directory="trainingsample",
|
| 88 |
+
commit_message="Optional commit message",
|
| 89 |
+
tags=["object-detection", "some_other_tag"],
|
| 90 |
+
include_optimizer=True
|
| 91 |
)
|
| 92 |
|
| 93 |
|