Add pipeline_tag, library_name, and additional tags to model card

#3
by nielsr HF Staff - opened

This PR improves the model card by adding key metadata:

  • pipeline_tag: image-to-image: This helps users discover the model when filtering by image-to-image tasks, which aligns with its core function of multi-task image restoration.
  • library_name: pytorch: The GitHub repository explicitly states it's a PyTorch implementation and shows PyTorch Lightning usage, allowing for automated code snippets if available and better library discoverability.
  • Additional tags: image-restoration, denoising, deraining, dehazing are added to categorize the specific tasks the model addresses, improving its discoverability on the Hub.
Tomk187 changed pull request status to merged

Sign up or log in to comment