Update README.md

README.md

@@ -113,11 +113,14 @@ from_pretrained_keras("your-username/your-awesome-model)
 			class="break-words leading-1 whitespace-pre-line text-xs md:text-sm text-gray-800">
 !pip install huggingface-hub
 !huggingface-cli login
+
 from transformers.keras_callbacks import PushToHubCallback
-from tensorflow.keras.callbacks import TensorBoard as TensorboardCallback
+from tensorflow.keras.callbacks import TensorBoard
 tensorboard_callback = TensorBoard(log_dir = "./logs/tensorboard)
 
-push_to_hub_callback = PushToHubCallback(output_dir="./logs", tokenizer=tokenizer,hub_model_id=model_id,)
+push_to_hub_callback = PushToHubCallback(output_dir="./logs", 
+										tokenizer=tokenizer,
+										hub_model_id=model_id)
 
 callbacks = [tensorboard_callback, push_to_hub_callback]
 model.fit(..., callbacks=callbacks, ...)