my-test-model / README.md
aungkomyint's picture
Upload folder using huggingface_hub
1eb0a70 verified
metadata
language: en
license: apache-2.0
tags:
  - text-classification
  - sentiment-analysis
  - test-model

My Test Sentiment Model

This is a test model for learning how to upload models to Hugging Face Hub.

Model Description

  • Base Model: DistilBERT
  • Task: Sentiment Analysis (Positive/Negative)
  • Language: English

Usage

from transformers import pipeline

classifier = pipeline("sentiment-analysis", model="YOUR_USERNAME/my-test-model")
result = classifier("I love this!")
print(result)

Training Data

This model was fine-tuned on the SST-2 dataset.