Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

dlb
/
MentalBERTa

Text Classification
Transformers
PyTorch
Safetensors
English
roberta
fill-mask
depression
medical
Model card Files Files and versions
xet
Community
1

Instructions to use dlb/MentalBERTa with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use dlb/MentalBERTa with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="dlb/MentalBERTa")
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForMaskedLM
    
    tokenizer = AutoTokenizer.from_pretrained("dlb/MentalBERTa")
    model = AutoModelForMaskedLM.from_pretrained("dlb/MentalBERTa")
  • Notebooks
  • Google Colab
  • Kaggle
MentalBERTa
2.84 GB
Ctrl+K
Ctrl+K
  • 3 contributors
History: 6 commits
ju-resplande's picture
ju-resplande
Update README.md
f55db5c verified 9 months ago
  • .gitattributes
    1.52 kB
    initial commit almost 3 years ago
  • README.md
    2.52 kB
    Update README.md 9 months ago
  • config.json
    710 Bytes
    Initial Commit almost 3 years ago
  • eval_results.txt
    59 Bytes
    Initial Commit almost 3 years ago
  • merges.txt
    456 kB
    Initial Commit almost 3 years ago
  • model.safetensors
    1.42 GB
    xet
    Adding `safetensors` variant of this model (#1) over 2 years ago
  • model_args.json
    2.85 kB
    Initial Commit almost 3 years ago
  • pytorch_model.bin
    1.42 GB
    xet
    Initial Commit almost 3 years ago
  • special_tokens_map.json
    957 Bytes
    Initial Commit almost 3 years ago
  • tokenizer_config.json
    1.31 kB
    Initial Commit almost 3 years ago
  • training_args.bin

    Detected Pickle imports (1)

    • "simpletransformers.config.model_args.LanguageModelingArgs"

    How to fix it?

    3.44 kB
    xet
    Initial Commit almost 3 years ago
  • vocab.json
    999 kB
    Initial Commit almost 3 years ago