Update trainer.py

trainer.py

@@ -13,7 +13,7 @@ from tqdm import tqdm
 sac = os.getenv('accesstoken')
 sn.set(font_scale=1.4)
 
-class_names = ['buildings', 'forest', 'glacier', 'mountain', 'sea', 'street']
+class_names = [ 'mountain', 'sea', 'street']
 class_names_label = {class_name: i for i, class_name in enumerate(class_names)}
 nb_classes = len(class_names)
 print(class_names_label)
@@ -93,7 +93,7 @@ model.save("model.keras")
 # Upload the model to your Hugging Face space repository
 push_to_hub_keras(
     model,
-    repo_id="okeowo1014/imgclassifiertraining",
+    repo_id="okeowo1014/imgclassifierb",
     commit_message="Optional commit message",
     tags=["image-classifier", "some_other_tag"],
     include_optimizer=True, token=sac