slachet/simple-nn-model

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:

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)