Update pipeline tag, add library name and GitHub link, fix example import

#1
by nielsr HF Staff - opened

This PR improves the model card for the Ariadne model by:

  • Correcting the pipeline_tag from reinforcement-learning to image-text-to-text, accurately reflecting the model's multimodal input and text generation capabilities.
  • Adding library_name: transformers to the metadata, which enables the automated "how to use" widget on the Hugging Face Hub, as the model demonstrates compatibility with the transformers library.
  • Including a direct link to the GitHub repository (https://github.com/Minghe-Shen/Ariadne) in the content for easier access to the associated code and project details.
  • Adding import torch to the sample usage code snippet, which is necessary for the snippet to execute correctly due to its use of torch_dtype and torch.cuda.is_available().
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment