Downloads · 30 days
11
9% of all-time downloads
peeyush01/bert-qa-finetuned
bert-qa-finetuned is a question answering model from peeyush01. Use it when the input is a question plus a passage. It is set up for transformers. The card lists the license as apache-2.0.
This model is a fine-tuned version of BERT-base uncased on the SQuAD v2 dataset for extractive question answering. It was trained for 3 epochs and can answer questions given a context passage, while also handling unan…
Downloads · 30 days
11
9% of all-time downloads
All-time downloads
122
Public
Parameters
109M
436 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors436 MB · 100%
From the Hugging Face model README
This model is a fine-tuned version of BERT-base uncased on the SQuAD v2 dataset for extractive question answering.
It was trained for 3 epochs and can answer questions given a context passage, while also handling unanswerable questions (a key feature of SQuAD v2).
You can try the model with the following code:
from transformers import pipeline
qa_pipeline = pipeline("question-answering", model="peeyush01/bert-squad-v2")
result = qa_pipeline({
"context": "Hugging Face is creating a tool that democratizes AI.",
"question": "What is Hugging Face creating?"
})
print(result)