Text Generation
PEFT
Safetensors
medical
biology
chemistry
conversational

Improve model card: Update pipeline tag, library, and add paper/repo details

#3
by nielsr HF Staff - opened

This PR significantly improves the model card for DrugReasoner by:

  • Updating the pipeline_tag from text-classification to text-generation to accurately reflect the model's capability to generate reasoning rationales alongside predictions. This ensures better discoverability on the Hugging Face Hub.
  • Changing the library_name from peft to transformers. As the model is a PEFT adapter built upon a transformers base model (Llama 3.1), transformers is the primary library for model interaction, enabling the automated "Use in Transformers" code snippet.
  • Correcting the arXiv badge text to display the accurate paper ID (2508.18579) while retaining the original arXiv link.
  • Adding an explicit GitHub badge for direct access to the model's code repository.
  • Including the full paper abstract to provide a detailed overview of the research.
  • Incorporating additional informative sections from the GitHub README, such as "Key Features", "Dataset & Model", "Performance", and a dedicated "License" section, to make the model card more comprehensive and user-friendly.

These updates aim to provide users with a more accurate and complete understanding of DrugReasoner.

Moreza009 changed pull request status to merged

Sign up or log in to comment