Model card suggestions 🙂
#1
by
merve
HF staff
- opened
Hello 👋
Did you use push_to_hub_keras to push the model to the Hub? If that's the case, there should be hyperparameters and other info generated for the model. See this model card as an example: https://huggingface.co/keras-io/deit at the bottom.
Hey @merve , Can you help on how to push a GAN model to the Hugging-Face Hub? Here is the code: https://colab.research.google.com/drive/1P0TXTvYXNVfcrwWYED1a_GVjwHhs_X-R?usp=sharing.
And yes, I will update the model card.
This line
push_to_hub_keras(discriminator,log_dir='logs/fit',repo_url = "https://huggingface.co/lucifertrj/DCGAN-to-generate-face-images",commit_message="model push to keras")
is right. You need to push generator only instead of discriminator, so you need to do:
push_to_hub_keras(generator,log_dir='logs/fit',repo_url = "https://huggingface.co/lucifertrj/DCGAN-to-generate-face-images",commit_message="model push to keras")
and that's it!
My bad, I will try again
lucifertrj
changed discussion status to
closed