Mobilebert Squadv2

🚀 Mobile-Bert fine-tuned on Squad V2 dataset

This model is based on the MobileBERT architecture, which is suitable for handheld devices or those with limited resources, providing efficient question-answering capabilities.

🚀 Quick Start

✨ Features

• Lightweight Architecture: Based on the MobileBERT architecture, it is suitable for devices with limited resources.
• Fine-tuned on Squad V2: Specifically optimized for question-answering tasks using the Squad V2 dataset.

📦 Installation

The installation mainly involves setting up the transformers library. You can install it via the following command:

pip install transformers

💻 Usage Examples

Basic Usage

# using transformers library first load model and Tokenizer
from transformers import AutoModelForQuestionAnswering,  AutoTokenizer, pipeline

model_name = "aware-ai/mobilebert-squadv2"

model = AutoModelForQuestionAnswering.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

Advanced Usage

# use question answering pipeline
qa_engine = pipeline('question-answering', model=model, tokenizer=tokenizer)
QA_input = {
    'question': 'your question?',
    'context': '. your context ................ '
}
res = qa_engine (QA_input)