update online

trainer.py

@@ -1,3 +1,4 @@
+from huggingface_hub import push_to_hub_keras
 import numpy as np
 import pandas as pd
 import os
@@ -67,11 +68,20 @@ model = tf.keras.models.Sequential([
 ])
 
 model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
-model.fit(train_images, train_labels, epochs=10, validation_split=0.1)
+model.fit(train_images, train_labels, epochs=2, validation_split=0.1)
 
 # Evaluate the model
 model.evaluate(test_images, test_labels)
 
 # save the model
 model.save("model.keras")
+push_to_hub_keras(model,
+    "okeowo1014/trainingsample",
+    "trainingsample",
+    "optional commit message"
+    include_optimizer = True,
+    tags = ["object-detection", "some_other_tag"],
+    **model_save_kwargs,
+)
+