transformers pytorch distilbert text-classification en license:mit autotrain_compatible endpoints_compatible region:us
---
language:
- en
library_name: transformers
license: mit
---
# Simple Neural Network Model

This is a simple neural network model designed for text classification tasks. The model uses a basic feedforward neural network architecture, trained for binary classification.

## Model Details

- **Architecture**: A simple feedforward neural network with one hidden layer.
- **Number of classes**: 2 (binary classification)
- **Input size**: 768 (or as per your dataset)

## Usage

To use this model with Hugging Face Transformers:

```python
from transformers import AutoModel, AutoTokenizer

# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("your-username/simple-nn-model")
model = AutoModel.from_pretrained("your-username/simple-nn-model")

# Tokenize inputs
inputs = tokenizer("Sample input text", return_tensors="pt")

# Get model predictions
outputs = model(**inputs)