okeowo1014 commited on
Commit
a2d41fa
·
1 Parent(s): 7945703

Update trainer.py

Browse files
Files changed (1) hide show
  1. 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(model,
79
- "okeowo1014/trainingsample",
80
- "trainingsample",
81
- "optional commit message",
82
- include_optimizer = True,
83
- tags = ["object-detection", "some_other_tag"],
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