Update trainer.py

trainer.py

@@ -68,7 +68,7 @@ model = tf.keras.models.Sequential([
 ])
 
 model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
-model.fit(train_images, train_labels, epochs=2, validation_split=0.1)
+model.fit(train_images, train_labels, epochs=10, validation_split=0.1)
 
 # Evaluate the model
 model.evaluate(test_images, test_labels)
@@ -83,9 +83,9 @@ model.save("model.keras")
 # Upload the model to your Hugging Face space repository
 push_to_hub_keras(
     model,
-    repo_id="okeowo1014/trainingsample",
+    repo_id="okeowo1014/imgclassifiertrainingsample",
     commit_message="Optional commit message",
-    tags=["image-detection", "some_other_tag"],
+    tags=["image-classifier", "some_other_tag"],
     include_optimizer=True,token=sac
 )