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_tagfromreinforcement-learningtoimage-text-to-text, accurately reflecting the model's multimodal input and text generation capabilities. - Adding
library_name: transformersto the metadata, which enables the automated "how to use" widget on the Hugging Face Hub, as the model demonstrates compatibility with thetransformerslibrary. - 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 torchto the sample usage code snippet, which is necessary for the snippet to execute correctly due to its use oftorch_dtypeandtorch.cuda.is_available().